runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/home/user/app/app.py", line 20, in <module> tumor_model.load_state_dict(renamed_sd) 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 TumorClassification: Missing key(s) in state_dict: "model.14.weight", "model.14.bias". Unexpected key(s) in state_dict: "model.8.weight", "model.8.bias". size mismatch for model.10.weight: copying a param with shape torch.Size([256, 512]) from checkpoint, the shape in current model is torch.Size([512, 86528]). size mismatch for model.10.bias: copying a param with shape torch.Size([256]) from checkpoint, the shape in current model is torch.Size([512]). size mismatch for model.12.weight: copying a param with shape torch.Size([4, 256]) from checkpoint, the shape in current model is torch.Size([256, 512]). size mismatch for model.12.bias: copying a param with shape torch.Size([4]) from checkpoint, the shape in current model is torch.Size([256]).

Container logs: