doesnot run train.py completely.

[upd-muku24]

Hi, can you please help me to run the code completely. whenever i try to run the train.py module, it produce following information and then stop. currently i am trying to run the code in Google colab.

The experiment name is stgan The experiment mode is train

---

directory created successfully [INFO]: Hi, This is root. [INFO]: After the configurations are successfully processed and dirs are created. [INFO]: The pipeline of the project will begin now. [INFO]: Creating STGAN architecture... [INFO]: Operation will be on GPU-CUDA [INFO]: __Python VERSION: 3.8.10 (default, Nov 14 2022, 12:59:47) [GCC 9.4.0] [INFO]: __PyTorch VERSION: 1.13.1+cu116 [INFO]: __CUDA VERSION cat: /usr/local/cuda/version.txt: No such file or directory [INFO]: __CUDNN VERSION: 8302 [INFO]: __Number CUDA Devices: 1 [INFO]: __Devices index, name, driver_version, memory.total [MiB], memory.used [MiB], memory.free [MiB] 0, Tesla T4, 510.47.03, 15360 MiB, 3 MiB, 15106 MiB [INFO]: __Active CUDA Device: GPU 0 [INFO]: __Available devices 1 [INFO]: __Current cuda device 0 Please wait while finalizing the operation.. Thank you