Open jiahui1014078 opened 4 months ago
Thanks for your interest in our work. 16G memory is too small, please set num_worker to 1 in the configuration file and then try to run it again. On the NYU dataset, our "teacher" network takes about 24 hours to train on a single 3090 GPU, while our "student" network takes about 10 hours to train on a single 3090 GPU.
My device is Nvidia RTX3090 GPU RAM 16, can I run this code? How much time does it take approximately? How much training time does it take with your configuration?