{
 "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
}