search

bluestyle97 / STGAN-pytorch

STGAN: A Unified Selective Transfer Network for Arbitrary Image Attribute Editing
89 stars 21 forks source link

doesnot run train.py completely. #13

Open upd-muku24 opened 1 year ago

upd-muku24 commented 1 year ago

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: 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

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