sjenni / temporal-ssl

Video Representation Learning by Recognizing Temporal Transformations. In ECCV, 2020.
https://sjenni.github.io/temporal-ssl/
GNU General Public License v3.0
48 stars 1 forks source link

Extremely Slow Training #6

Open kennymckormick opened 3 years ago

kennymckormick commented 3 years ago

Thanks for releasing the temporal-ssl codebase. I run the temporal-ssl training with command python train_test_C3D.py --tag='test' --num_gpus=2 and find that the training is extremely slow on a 8-TitanXp server: it takes several minutes for one iteration:

image

How long does it take to train temporal-ssl on your platform? Would you please share the training log?

sjenni commented 3 years ago

Hi, Sorry for the late response. Is the issue already resolved? If not, are you sure that tensorflow sees/uses the GPUs.