Closed cjl506 closed 9 months ago
测试的时候,提示这样的错误,该如何解决呢
你好,我在测试时没有发现该问题,你可以尝试在主函数里用以下两句,而不调用‘load_checkpoint’函数运行一下: checkpoint = torch.load($weights_path$, map_location=device) model.load_state_dict(checkpoint)