KeyError: 'epoch' on running track.py with hrnetv2_w32_imagenet_pretrained.pth

[adityamd]

Hello,
I want to use the code with pretrained imagenet weights: hrnetv2_w32_imagenet_pretrained.pth ( given as the official fairmot implementation on imagenet dataset). However, running the command:

python track.py mot --load_model /content/FairMOT/models/hrnetv2_w32_imagenet_pretrained.pth --conf_thres 0.4 --data_dir /content/FairMOT --val_mot20 VAL_MOT20

gives me the following error:

Fix size testing.
training chunk_sizes: [6, 6]
The output will be saved to  /content/FairMOT/src/lib/../../exp/mot/default
heads {'hm': 1, 'wh': 2, 'id': 512, 'reg': 2}
2021-10-11 05:01:30 [INFO]: start seq: MOT20-01
Creating model...
Traceback (most recent call last):
  File "track.py", line 240, in <module>
    save_videos=False)
  File "track.py", line 112, in main
    save_dir=output_dir, show_image=show_image, frame_rate=frame_rate)
  File "track.py", line 56, in eval_seq
    tracker = JDETracker(opt, frame_rate=frame_rate)
  File "/content/FairMOT/src/lib/tracker/multitracker.py", line 181, in __init__
    self.model = load_model(self.model, opt.load_model)
  File "/content/FairMOT/src/lib/models/model.py", line 37, in load_model
    print('loaded {}, epoch {}'.format(model_path, checkpoint['epoch']))
KeyError: 'epoch'

Any help will be appreciated. Thank you