yunxiaoshi / Neural-IMage-Assessment

A PyTorch Implementation of Neural IMage Assessment
Other
508 stars 92 forks source link

Update main.py #5

Closed George3d6 closed 5 years ago

George3d6 commented 5 years ago

Move some training specific stuff under training, so this can actually be ran in test mode without valid values for those args