forward_train() missing 1 required positional argument: 'gt_label'

[liuxy2318]

When I run the mmclassification prediction demo, I get the correct prediction results, but when exporting the onnx model with torch.onnx.export(model, x, "resnet50.onnx", verbose=True), I always report an error. This error is the same as when I tried to export a model suitable for invocation in C++ using traced_module=torch.jit.trace(net, x).

[mzr1996]

Which version are you using?
If you are using mmcls 0.x, we have an onnx export script here
If you are using mmcls 1.x, we have moved our deploy-related functionality to MMDeploy, welcome to use it!

[Songhuilun]

Hello，can you solve this problem?