filter-grafting icon indicating copy to clipboard operation
filter-grafting copied to clipboard

关于Grafting.py中的几个问题,盼复

Open yuanpanlifly opened this issue 1 year ago • 0 comments

176行 torch.save(state, '%s/ckpt%d_%d.t7' % (args.s, args.i % args.num, epoch)),保存了一个epoch训练好的模型权重 106行 checkpoint = torch.load('%s/ckpt%d_%d.t7' % (args.s, args.i - 1, epoch))['net']。加载已有的权重信息,可是没有这个名字的权重信息,会导致程序在 time.sleep(10)死循环里,希望能够得到您的解答感谢

yuanpanlifly avatar Jan 09 '24 10:01 yuanpanlifly