sunshineatnoon/PytorchWCT

models given don't work on Windows

jnnnnn opened this issue · 3 comments

Torch binary serialization is platform-dependent which means that the given models don't work in Windows.

Hi, I haven't tried to run torch/pytorch on Windows, maybe you can ask on Pytorch Forum or Torch Googlegroup.

Get hold of a linux machine, use this code to convert to .pth files. Use them.

use pytorch 0.4 version (torch legacy + serialization load lua function) and from utill.py, add argument on load_lua such as
vgg1 = load_lua(args.vgg1,long_size=8) because window is reading it as 4 bit

on my utill.py
class WCT(nn.Module):
def init(self,args):
super(WCT, self).init()

    # load pre-trained network
     vgg1 = load_lua(args.vgg1,long_size=8)
    decoder1_torch = load_lua(args.decoder1,long_size=8)
    vgg2 = load_lua(args.vgg2,long_size=8)
    decoder2_torch = load_lua(args.decoder2,long_size=8)
    vgg3 = load_lua(args.vgg3,long_size=8)
    decoder3_torch = load_lua(args.decoder3,long_size=8)
    vgg4 = load_lua(args.vgg4,long_size=8)
    decoder4_torch = load_lua(args.decoder4,long_size=8)
    vgg5 = load_lua(args.vgg5,long_size=8)
    decoder5_torch = load_lua(args.decoder5,long_size=8)