100 lines
2.8 KiB
Plaintext
100 lines
2.8 KiB
Plaintext
{
|
|
"cells": [
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "73ce6820-8bcf-48bc-852d-41c5a6b7a440",
|
|
"metadata": {
|
|
"collapsed": false,
|
|
"is_executing": true,
|
|
"ExecuteTime": {
|
|
"start_time": "2025-04-18T06:58:57.940747Z"
|
|
}
|
|
},
|
|
"outputs": [
|
|
{
|
|
"name": "stdout",
|
|
"output_type": "stream",
|
|
"text": [
|
|
"加载 PEMSD3 数据集中... \r\n",
|
|
"Creat Log File in: /home/czzhangheng/code/TrafficWheel/experiments/PEMSD3/2025-04-18_14-59-15/run.log\r\n",
|
|
"04/18 14:59: Experiment log path in: /home/czzhangheng/code/TrafficWheel/experiments/PEMSD3/2025-04-18_14-59-15\r\n",
|
|
"04/18 14:59: Training process started\r\n",
|
|
"Train Epoch 1: 100%|███████████████| 245/245 [00:08<00:00, 29.08it/s, loss=19.6]\r\n",
|
|
"04/18 14:59: Train Epoch 1: average Loss: 37.727569, time: 8.43 s\r\n",
|
|
"Val Epoch 1: 100%|██████████████████| 81/81 [00:00<00:00, 152.50it/s, loss=18.4]\r\n",
|
|
"04/18 14:59: Val Epoch 1: average Loss: 19.857144, time: 0.53 s\r\n",
|
|
"Test Epoch 1: 100%|█████████████████| 82/82 [00:00<00:00, 157.36it/s, loss=20.9]\r\n",
|
|
"04/18 14:59: Test Epoch 1: average Loss: 19.833939, time: 0.52 s\r\n",
|
|
"04/18 14:59: Best validation model saved!\r\n",
|
|
"Train Epoch 2: 18%|██▊ | 44/245 [00:01<00:06, 30.37it/s, loss=21.2]"
|
|
]
|
|
}
|
|
],
|
|
"source": [
|
|
"!python run.py --model EXP --dataset PEMSD3 --mode train --device cuda:1"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "413ac989e85adf6a",
|
|
"metadata": {
|
|
"collapsed": false,
|
|
"is_executing": true
|
|
},
|
|
"outputs": [],
|
|
"source": []
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "ba12f992-f8a5-418d-a136-a4e1bdf0a3f3",
|
|
"metadata": {
|
|
"collapsed": false,
|
|
"scrolled": true,
|
|
"is_executing": true
|
|
},
|
|
"outputs": [],
|
|
"source": [
|
|
"!python run.py --model EXP --dataset PEMSD8 --mode train --device cuda:1"
|
|
]
|
|
},
|
|
{
|
|
"cell_type": "code",
|
|
"execution_count": null,
|
|
"id": "68bfc782-42a1-4d3e-aea6-8ca6d24e15a1",
|
|
"metadata": {
|
|
"collapsed": false,
|
|
"scrolled": true,
|
|
"is_executing": true
|
|
},
|
|
"outputs": [],
|
|
"source": [
|
|
"!python run.py --model EXP --dataset PEMSD7 --mode train --device cuda:1"
|
|
]
|
|
}
|
|
],
|
|
"metadata": {
|
|
"kernelspec": {
|
|
"display_name": "Python 3 (ipykernel)",
|
|
"language": "python",
|
|
"name": "python3"
|
|
},
|
|
"language_info": {
|
|
"codemirror_mode": {
|
|
"name": "ipython",
|
|
"version": 3
|
|
},
|
|
"file_extension": ".py",
|
|
"mimetype": "text/x-python",
|
|
"name": "python",
|
|
"nbconvert_exporter": "python",
|
|
"pygments_lexer": "ipython3",
|
|
"version": "3.10.16"
|
|
}
|
|
},
|
|
"nbformat": 4,
|
|
"nbformat_minor": 5
|
|
}
|