runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/home/user/app/app.py", line 23, in <module> model.load_state_dict(torch.load(MODEL_PATH, map_location=torch.device('cpu'))) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2593, in load_state_dict raise RuntimeError( RuntimeError: Error(s) in loading state_dict for SimpleCNN: Missing key(s) in state_dict: "conv1.weight", "conv1.bias", "conv2.weight", "conv2.bias", "fc1.weight", "fc1.bias", "fc2.weight", "fc2.bias". Unexpected key(s) in state_dict: "model_state_dict", "config", "training_metrics", "dataset_info".

Container logs: