cnn-ml.py 属性错误 train_kd.py model问题

[QISIRUI]

请教大佬这个bug应该怎么解决呢

另外，在train_kd.py中，定义优化器与损失函数那里，model是标红的，直接运行，显示没有定义；

当用teacher_model,student_model代替model时，显示下面这个bug。

optimizer = optim.Adam(filter(lambda p: p.requires_grad, model.parameters()), lr=cfg.LR) optimizer = optim.SGD(model.parameters(), lr=cfg.LR, momentum=cfg.MOMENTUM, weight_decay=cfg.WEIGHT_DECAY)

抱拳

[lxztju]

对于第一个问题，你需要到torchvision的模型定义中，提取你需要的模型层的输出，我这里采用的时efficientnet，为了省事没有去处理别的模型，在程序文件的注释中已经指明了这个问题 对于第二个问题，这是因为img放置在了gpu上，而模型忘记了，这个在程序中已经修改过来，谢谢指出

[LifeIsBright-heihei]

你好，请问你解决第一个问题了吗，如果用resnet的话，要怎么做呢