PytorchWCT

PytorchWCT copied to clipboard

Fix: Read Lua weights with Pytorch > 1.0

2

from torch.utils.serialization import load_lua doesnt work in current pytorch versions here is a possible fix with **torchfile** ``` class pytorch_lua_wrapper: def __init__(self, lua_path): self.lua_model = torchfile.load(lua_path) def get(self, idx): return...

ariel415el

Make the program runs well in Windows and Pytorch 1.3+.

S-HuaBomb

Issues with SVD regularization

4

Hi, first off: thanks for the concise and easy to follow implementation, and congrats for the work building on it, I really enjoyed it. :) I `util.py` you write: https://github.com/sunshineatnoon/PytorchWCT/blob/2a2b4e490394272665ced89634290e63314872d8/util.py#L49...

black-puppydog

Migrate to Pytorch 1.0

2

This takes the code all the way to pytorch 1.0: I had to migrate the encoder/decoder models since `load_lua` seems to have been removed from the API (?!) and while...

black-puppydog

May I ask how to generate the Inverting white features for Figure 2,thanks a lot！ 

ZzzTD

thenotshyshyshy

KeyError: 'torch.CudaTensor'

2

Hi，I met KeyError: 'torch.CudaTensor' when running "python3 WCT.py --cuda". That's why? Look forward for your reply.

dmmSJTU