Training time

[markNZed]

Training time

[zezhishao]

Thanks for the reminder, this was manually interrupted by myself because I think the training epoch is enough at that point. However, training for a few more epochs should continue to improve performance.

[zezhishao]

Thank you for your attention.
You can find my training logs in training_logs/.
Specifically, the TSFormer takes about 500 seconds/epoch on a single 3090 GPU.
On two 3090 GPUs, the whole STEP takes about 500 seconds/epoch.
You can use more GPUs by simply modifying CFG.NUM_GPU in the configuration file.

[markNZed]

That is great, thanks. I should have seen the training logs. I noticed one of them did not run through to completion:

STEP/training_logs/TSFormer_PEMS04.log

Line 960 in ea2c3c3

2022-09-29 10:04:47,922 - easytorch-training - ERROR - Traceback (most recent call last):

Best wishes for your research.