search

cyclomon / 3dbraingen

Official Pytorch Implementation of "Generation of 3D Brain MRI Using Auto-Encoding Generative Adversarial Network" (accepted by MICCAI 2019)
MIT License
126 stars 38 forks source link

In Test.ipynb, `image = gen_load.__next__()` where gen_load is undefined? #4

Closed LUYU0004 closed 1 year ago

cyclomon commented 4 years ago

Sorry, there was a mistake. after declaring inf_train_gen ( code for infinite data loader ), there should have been a code "gen_load = inf_train_gen(train_loader)" I included this line.

Thanks.

LUYU0004 commented 4 years ago

Thank you for your reply! And there is another minor issue: in pytorch_ssim.py, the filters variable in create_window(..) function is undefined.