躺不平的废柴 2022-10-06 16:28 采纳率: 75%
浏览 34
已结题

谁能帮我看看这个报错是什么意思哇?

遇到报错AssertionError


Successfully loaded first 20 categories for training and last 20 for testing!
Traceback (most recent call last):
  File "D:\d.py", line 1438, in _exec
    pydev_imports.execfile(file, globals, locals)  # execute the script
  File "D:\e.py", line 18, in execfile
    exec(compile(contents+"\n", file, 'exec'), glob, loc)
  File "E:/t.py", line 349, in <module>
    main()
  File "E:/t.py", line 298, in main
    _init_(args)
  File "E:/t.py", line 339, in _init_
    assert os.path.isfile(args.pretrained1)
AssertionError

部分代码

def _init_(args):
    if not os.path.exists('checkpoints'):
        os.makedirs('checkpoints')
    if not os.path.exists('checkpoints/' + args.exp_name):
        os.makedirs('checkpoints/' + args.exp_name)
    if not os.path.exists('checkpoints/' + args.exp_name + '/' + 'models'):
        os.makedirs('checkpoints/' + args.exp_name + '/' + 'models')
    trainset = RegistrationData(ModelNet40Data(train=True, num_points=args.num_points, unseen=args.unseen),
                                partial_source=args.partial_source, noise=args.noise, outliers=args.outliers)
    testset = RegistrationData(ModelNet40Data(train=False, num_points=args.num_points, unseen=args.unseen),
                                partial_source=args.partial_source, noise=args.noise, outliers=args.outliers)
    train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
    test_loader = DataLoader(testset, batch_size=args.test_batch_size, shuffle=False, drop_last=False, num_workers=args.workers)

    if not torch.cuda.is_available():
        args.device = 'cpu'
    args.device = torch.device(args.device)

    model = tpccNet()
    model = model.to(args.device)

    ptnet = PointNet(emb_dims=args.emb_dims)
    model2 = itmpeNet(feature_model=ptnet)
    model2 = model2.to(args.device)

    checkpoint = None
    if args.resume:
        assert os.path.isfile(args.resume)
        checkpoint = torch.load(args.resume)
        args.start_epoch = checkpoint['epoch']        #Here checkpoint is used to call the saved model to continue training

        model2.load_state_dict(checkpoint['model2'])

    if args.pretrained1:
        assert os.path.isfile(args.pretrained1)     #这里报错
        model.load_state_dict(torch.load(args.pretrained1, map_location='cpu'))
    model.to(args.device)

    if args.eval:
        test(args, model, model2, test_loader, textio)
    else:
        train(args, model, model2, train_loader, test_loader, boardio, textio, checkpoint)

  • 写回答

1条回答 默认 最新

  • 天元浪子 Python领域优质创作者 2022-10-06 17:14
    关注

    这是断言错误抛出的异常。断言那一行,期望args.pretrained1是一个文件,期望落空了,因此抛出异常。如果没有这一行,估计下一行会出错,因为args.pretrained1不是一个文件,torch.load也无法正常执行。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 10月14日
  • 已采纳回答 10月6日
  • 创建了问题 10月6日

悬赏问题

  • ¥15 装 pytorch 的时候出了好多问题,遇到这种情况怎么处理?
  • ¥20 IOS游览器某宝手机网页版自动立即购买JavaScript脚本
  • ¥15 手机接入宽带网线,如何释放宽带全部速度
  • ¥30 关于#r语言#的问题:如何对R语言中mfgarch包中构建的garch-midas模型进行样本内长期波动率预测和样本外长期波动率预测
  • ¥15 ETLCloud 处理json多层级问题
  • ¥15 matlab中使用gurobi时报错
  • ¥15 这个主板怎么能扩出一两个sata口
  • ¥15 不是,这到底错哪儿了😭
  • ¥15 2020长安杯与连接网探
  • ¥15 关于#matlab#的问题:在模糊控制器中选出线路信息,在simulink中根据线路信息生成速度时间目标曲线(初速度为20m/s,15秒后减为0的速度时间图像)我想问线路信息是什么