LightModels icon indicating copy to clipboard operation
LightModels copied to clipboard

Published 20 hours ago • verified publisher icon Marcovaldong

some errors

Open bigdaronlee163 opened this issue 5 years ago • 1 comments

in the method :test of main.py ,net.load_state_dict('./checkpoint/{}.pth'.format(args.model)) cannot run ,but net.load_state_dict(torch.loda('./checkpoint/{}.pth'.format(args.model))) can solve this problem.

bigdaronlee163 avatar Apr 24 '19 07:04 bigdaronlee163

my platform is pycharm and python3.7,pytorch 1.0.1

bigdaronlee163 avatar Apr 24 '19 07:04 bigdaronlee163