Small fixes to the UI and to keep up to date with pytorch changes.#12

Open

jmmcd wants to merge 4 commits intoznxlwm:masterfrom jmmcd:master

+65-46

Commits on May 23, 2018

Check CUDA is available if specified, and fix the --gpu_mode flag to allow user to turn it off.
jmmcd
committed
Add missing semi-colon in discriminator forward sigmoid array range. Fixes a warning about shape (UserWarning: Using a target size (torch.Size([64, 1])) that is different to the input size (torch.S…
jmmcd
committed
Replace data[0] with data.item() throughout the repo. Fixes a warning (UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tenso…
jmmcd
committed
Replace data[0] with data.item() in infoGAN (should have been in previous commit ef8de39). Fixes a warning (UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use t…
jmmcd
committed