Error with KITTI dataset.

[csj777]

Hi, thanks for your amazing work. We are trying to run it with KITTI dataset. Howerer, we meet some errors now.
we run like this:
python test.py --gpu 0 --workdir ./output --model grnet --weights ./models/GRNet-KITTI.pth --test_mode kitti

return

INFO: - Finish building dataset.
DEBUG: - Parameters in net_G: 76707626.
INFO: - Recovering from ./models/GRNet-KITTI.pth ...
Traceback (most recent call last):
File "test.py", line 80, in
main()
File "test.py", line 74, in main
model = getattr(module, args.model + "Runner")(cfg, logger)
File "/space2/home/chensj/projects/SpareNet/runners/grnet_runner.py", line 19, in init
super().init(config, logger)
File "/space2/home/chensj/projects/SpareNet/runners/base_runner.py", line 79, in init
self.models_load()
File "/space2/home/chensj/projects/SpareNet/runners/base_runner.py", line 108, in models_load
self.init_epoch, self.best_metrics = um.model_load(self.config, self.models)
File "/space2/home/chensj/projects/SpareNet/utils/misc.py", line 75, in model_load
net_G.load_state_dict(checkpoint["net_G"]) # change into net_G!!
KeyError: 'net_G'

Could you please give me some advice?