minor fix for hubconf - Githubissues

facebookresearch / pytorch_GAN_zoo

A mix of GAN implementations including progressive growing

BSD 3-Clause "New" or "Revised" License

1.62k stars 271 forks source link

minor fix for hubconf #33

Closed ailzhang closed 5 years ago

ailzhang commented 5 years ago

A few minor fix to get hubconf local test running.

To test locally:

Clone and rename this repo to ~/.torch/hub/pytorch_GAN_zoo_REFACTOR_hubconf

run with pytorch 1.0+

import torch.hub
torch.hub.load('fairinternal/pytorch_GAN_zoo:REFACTOR_hubconf', 'DCGAN',useGPU=True, pretrained=False)
#pretrained=True is waiting for pretrained weights to be available.