search

maudzung / SFA3D

Super Fast and Accurate 3D Object Detection based on 3D LiDAR Point Clouds (The PyTorch implementation)
https://github.com/maudzung/Super-Fast-Accurate-3D-Object-Detection
MIT License
1k stars 271 forks source link

KeyError: 'optimizer' when resuming training from checkpoint #48

Closed yukaryote closed 2 years ago

yukaryote commented 3 years ago

I am trying to resume training from the checkpoint provided in the repo (fpn_resnet_18_epoch_300.pth), and when I run python train.py --gpu_idx 0 --resume_path ../checkpoints/fpn_resnet_18/fpn_resnet_18_epoch_300.pth I get this error:

Traceback (most recent call last): File "train.py", line 281, in <module> main() File "train.py", line 67, in main main_worker(configs.gpu_idx, configs) File "train.py", line 132, in main_worker optimizer.load_state_dict(utils_state_dict['optimizer']) KeyError: 'optimizer'

maudzung commented 2 years ago

Hi @yukaryote I saved weights & optimizer states in 2 files

mouawiya commented 2 years ago

how do you resume from a checkpoint using 2 files?