Hangz-nju-cuhk / Rotate-and-Render

Code for Rotate-and-Render: Unsupervised Photorealistic Face Rotation from Single-View Images (CVPR 2020)

Creative Commons Attribution 4.0 International

489 stars 112 forks source link

Closed v-prgmr closed 3 years ago

v-prgmr commented 3 years ago

While running the train script, the dataloader gets stuck at data = training_queue.get(block=True) in train.py

OS: Linux 18.04

GPU: 4

Pytorch: 1.7.1 cuda CUDA: 11.0 cuDNN: 8.2