runtime error

Exit code: 1. Reason: h [22/45] (48.89%), Loss: 6.3548 Epoch [5/5], Batch [23/45] (51.11%), Loss: 6.3527 Epoch [5/5], Batch [24/45] (53.33%), Loss: 6.3487 Epoch [5/5], Batch [25/45] (55.56%), Loss: 6.3447 Epoch [5/5], Batch [26/45] (57.78%), Loss: 6.3419 Epoch [5/5], Batch [27/45] (60.00%), Loss: 6.3397 Epoch [5/5], Batch [28/45] (62.22%), Loss: 6.3363 Epoch [5/5], Batch [29/45] (64.44%), Loss: 6.3323 Epoch [5/5], Batch [30/45] (66.67%), Loss: 6.3295 Epoch [5/5], Batch [31/45] (68.89%), Loss: 6.3255 Epoch [5/5], Batch [32/45] (71.11%), Loss: 6.3236 Epoch [5/5], Batch [33/45] (73.33%), Loss: 6.3211 Epoch [5/5], Batch [34/45] (75.56%), Loss: 6.3181 Epoch [5/5], Batch [35/45] (77.78%), Loss: 6.3160 Epoch [5/5], Batch [36/45] (80.00%), Loss: 6.3129 Epoch [5/5], Batch [37/45] (82.22%), Loss: 6.3100 Epoch [5/5], Batch [38/45] (84.44%), Loss: 6.3073 Epoch [5/5], Batch [39/45] (86.67%), Loss: 6.3042 Epoch [5/5], Batch [40/45] (88.89%), Loss: 6.3016 Epoch [5/5], Batch [41/45] (91.11%), Loss: 6.2997 Epoch [5/5], Batch [42/45] (93.33%), Loss: 6.2974 Epoch [5/5], Batch [43/45] (95.56%), Loss: 6.2951 Epoch [5/5], Batch [44/45] (97.78%), Loss: 6.2917 Epoch [5/5], Batch [45/45] (100.00%), Loss: 6.2883 Epoch [5/5] completed. Avg Loss: 6.2883, Time: 201.47 seconds Training completed. Total Time: 462.14 seconds Traceback (most recent call last): File "/app/app.py", line 185, in <module> main() File "/app/app.py", line 176, in main save_model(model, MODEL_SAVE_PATH) File "/app/app.py", line 104, in save_model torch.save(model.state_dict(), path) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 651, in save with _open_zipfile_writer(f) as opened_zipfile: File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 525, in _open_zipfile_writer return container(name_or_buffer) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 496, in __init__ super().__init__(torch._C.PyTorchFileWriter(self.name)) RuntimeError: File simple_ai_model.pth cannot be opened.

Container logs: