Why doesn’t this work?
import torch
from collections import OrderedDict


model = torch.load('vgg-16.pth')


new_state_dict = OrderedDict()
for (k, v) in model.items():
            t = torch.ones(v.size()).float() *255
            v = v.mul(t.cuda())
            new_state_dict[k] = v

torch.save(new_state_dict, 'vgg-16-2.pth')