Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

small fixes for latest PyTorch & Windows (?) #16

Merged
merged 2 commits into from
May 19, 2020

Conversation

stnoh
Copy link
Contributor

@stnoh stnoh commented Apr 8, 2020

I fixed minor issues in the latest code and certificated that it works in windows.
Here are my testing environments.
Both are windows 10 machines and Anaconda3 (Anaconda3-2020.02-Windows-x86_64.exe) was used for Python3 installation.

  1. laptop setting
cudatoolkit==9.0
pillow=6.2.1
pytorch==1.1.0
torchvision==0.3.0
  1. desktop setting
cudatoolkit==10.1.243
pillow=6.2.1
pytorch==1.4.0
torchvision==0.5.0

I am not sure the second commit is working with old PyTorch (<1.x.x), but the first commit is still useful for general users.

stnoh added 2 commits April 8, 2020 21:30
- vgg19_model.py: independent to PyTorch version (.data[0] for PyTorch<=0.4.1 and .data.item() for PyTorch>0.4.1)
- MLS.py: more checking for pixel coordinate (0 <= x and 0 <= y)
- vgg19_model.py: size_average=True -> reduction='mean'
- util.py: Scale() -> Resize()
@stnoh stnoh mentioned this pull request Apr 8, 2020
@kfiraberman kfiraberman merged commit ada3fe4 into kfiraberman:master May 19, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants