Vinayak

icon location Magdeburg, Saxony-Anhalt icon location Deep Learning Researcher, Currently pursuing masters from Otto-von-Guericke Universität. This repo contains self projects I work on.

Results 2 comments of Vinayak

RuntimeError: Attempted to set the storage of a tensor on device "cuda:0" to a storage on different device "cpu".

Could you try modifying this line: state_dict = modelzoo.load_url(resnet18_url) to state_dict = modelzoo.load_url(resnet18_url, map_location=lambda storage, loc: storage.cuda())

is the beta pre-trained model better than the FOMM?

I think you could also try with LSGAN