clovaai / stargan-v2

StarGAN v2 - Official PyTorch Implementation (CVPR 2020)
Other
3.48k stars 653 forks source link

Update fid.py and eval.py #124

Open Yunyung opened 2 years ago

Yunyung commented 2 years ago

Hi 👋,

We think the num_workers parameter for DataLoader is important in some environments such as docker or limited RAM, otherwise causing failure. So we add num_workers when evaluating.


Supplementary materials

For example: In docker environment, the shared memory is limited causing specific error:

RuntimeError: DataLoader worker (pid xxxx) is killed by signal: Bus error. It is possible that dataloader's workers are out of shared memory. Please try to raise your shared memory limit.

References: https://github.com/pytorch/pytorch/issues/5040 https://github.com/ultralytics/yolov3/issues/1151