runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/home/user/app/app.py", line 12, in <module> generator.load_state_dict(torch.load("generator128.pth", map_location=device)) 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 Generator: Missing key(s) in state_dict: "main.0.weight", "main.1.weight", "main.1.bias", "main.1.running_mean", "main.1.running_var", "main.3.weight", "main.4.weight", "main.4.bias", "main.4.running_mean", "main.4.running_var", "main.6.weight", "main.7.weight", "main.7.bias", "main.7.running_mean", "main.7.running_var", "main.9.weight", "main.10.weight", "main.10.bias", "main.10.running_mean", "main.10.running_var", "main.12.weight". Unexpected key(s) in state_dict: "model.0.0.weight", "model.0.1.weight", "model.0.1.bias", "model.0.1.running_mean", "model.0.1.running_var", "model.0.1.num_batches_tracked", "model.1.0.weight", "model.1.1.weight", "model.1.1.bias", "model.1.1.running_mean", "model.1.1.running_var", "model.1.1.num_batches_tracked", "model.2.0.weight", "model.2.1.weight", "model.2.1.bias", "model.2.1.running_mean", "model.2.1.running_var", "model.2.1.num_batches_tracked", "model.3.0.weight", "model.3.1.weight", "model.3.1.bias", "model.3.1.running_mean", "model.3.1.running_var", "model.3.1.num_batches_tracked", "model.4.0.weight", "model.4.1.weight", "model.4.1.bias", "model.4.1.running_mean", "model.4.1.running_var", "model.4.1.num_batches_tracked", "model.5.weight", "model.5.bias".

Container logs: