Question about time of training

m-hasan-n / pooling

"Pooling Toolbox" is the code of our work "Maneuver-Aware Pooling for Vehicle Trajectory Prediction".

MIT License

51 stars 8 forks source link

Open liuyueChang opened 11 months ago

liuyueChang commented 11 months ago

Thank you for your work, what machine the model was trained on and how long it has been trained ?

harryguantj commented 9 months ago

Time duration under my devices. Hope to help:)

CPU: AMD EPYC 7542 32-Core Processor GPU: NVIDIA GeForce RTX 4090

Preprocess: MatlabR2018b Linux Parallel Pool: 6 workers 2024.02.03 16:26:06 - 2024.02.03 17:00:59

Train: Pycharm Professional Edition 2021.3 pretrain-epoch: 5 train-epoch: 3 2024.02.03 17:12:42 - 2024.02.03 18:59:13

Evaluate: Pycharm Professional Edition 2021.3 2024.02.03 19:01:46 - 2024.02.03 19:21:33