runtime error

Exit code: 1. Reason: Traceback (most recent call last): File "/app/app.py", line 11, in <module> G.load_state_dict(torch.load("generator.pth", map_location=DEVICE)) ~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/torch/nn/modules/module.py", line 2635, in load_state_dict raise RuntimeError( ...<3 lines>... ) RuntimeError: Error(s) in loading state_dict for Generator: Missing key(s) in state_dict: "net.0.weight", "net.0.bias", "net.0.running_mean", "net.0.running_var", "net.2.weight", "net.2.bias", "net.3.weight", "net.3.bias", "net.3.running_mean", "net.3.running_var", "net.6.weight", "net.6.bias", "net.7.weight", "net.7.bias", "net.7.running_mean", "net.7.running_var", "net.10.weight", "net.10.bias", "net.11.weight", "net.11.bias", "net.11.running_mean", "net.11.running_var", "net.14.weight", "net.14.bias". Unexpected key(s) in state_dict: "main.0.weight", "main.0.bias", "main.1.weight", "main.1.bias", "main.1.running_mean", "main.1.running_var", "main.1.num_batches_tracked", "main.3.weight", "main.3.bias", "main.4.weight", "main.4.bias", "main.4.running_mean", "main.4.running_var", "main.4.num_batches_tracked", "main.6.weight", "main.6.bias", "main.7.weight", "main.7.bias", "main.7.running_mean", "main.7.running_var", "main.7.num_batches_tracked", "main.9.weight", "main.9.bias", "main.10.weight", "main.10.bias", "main.10.running_mean", "main.10.running_var", "main.10.num_batches_tracked", "main.12.weight", "main.12.bias". size mismatch for fc.weight: copying a param with shape torch.Size([16384, 100]) from checkpoint, the shape in current model is torch.Size([8192, 256]). size mismatch for fc.bias: copying a param with shape torch.Size([16384]) from checkpoint, the shape in current model is torch.Size([8192]).

Container logs: