runtime error

0%| | 0.00/107M [00:00<?, ?B/s] 100%|██████████| 107M/107M [00:00<00:00, 346MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 28, in <module> model.load_state_dict(checkpoint['model_state_dict']) KeyError: 'model_state_dict'

Container logs:

Fetching error logs...