Skip to content

pyTorch upgrade 0.3.1 -> 1.1.0 (incl. trained weights)#21

Open
dgrechka wants to merge 2 commits intozoogzog:masterfrom
dgrechka:torch.1.1.0
Open

pyTorch upgrade 0.3.1 -> 1.1.0 (incl. trained weights)#21
dgrechka wants to merge 2 commits intozoogzog:masterfrom
dgrechka:torch.1.1.0

Conversation

@dgrechka
Copy link

  • Upgrading the code to be compatible with pyTorch 1.1.0 (previously compatible with pyTorch 0.3.1 only)

  • The trained weights are updated (now is to be loaded in pyTorch 1.1.0 or newer).

This pull request resolves #7 (new trained weights file is compatible with recent pyTorch) and resolves #19 (code updated)

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.

CUDA Memory Error Unable to extract m-25012018-123527.pth.tar

1 participant