microsoft / Bringing-Old-Photos-Back-to-Life

Bringing Old Photo Back to Life (CVPR 2020 oral)
https://arxiv.org/abs/2004.09484
MIT License
15.15k stars 2k forks source link

Training in colab (train_domain_A) #206

Open SofiaVega opened 3 years ago

SofiaVega commented 3 years ago

Hello, We are trying to train domain A in a colab notebook, and we keep getting an error where the dataset size is zero, even though our training images are being loaded into the .bigfile files and into the UnPairOldPhotos_SR dataset. Do we need to change something from the data loader classes in order to train this model in a colab notebook?