runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/app/app.py", line 84, in <module> model.load_state_dict(state_dict) File "/usr/local/lib/python3.11/site-packages/torch/nn/modules/module.py", line 2581, in load_state_dict raise RuntimeError( RuntimeError: Error(s) in loading state_dict for SleepStageCNN: Missing key(s) in state_dict: "network.1.weight", "network.1.bias", "network.1.running_mean", "network.1.running_var", "network.4.weight", "network.4.bias", "network.5.weight", "network.5.bias", "network.5.running_mean", "network.5.running_var", "network.9.weight", "network.9.bias", "network.12.weight", "network.12.bias". Unexpected key(s) in state_dict: "network.3.weight", "network.3.bias", "network.7.weight", "network.7.bias", "network.10.weight", "network.10.bias". size mismatch for network.0.weight: copying a param with shape torch.Size([32, 2, 50]) from checkpoint, the shape in current model is torch.Size([32, 1, 50]).

Container logs: