Error running training (LR Scheduler) #59
Description
The command I ran:
python3 scripts/train.py +experiment=cvt_nuscenes_vehicle data.dataset_dir=/media/datasets/nuscenes data.labels_dir=/media/datasets/cvt_labels_nuscenes
The errors I got:
Error executing job with overrides: ['+experiment=cvt_nuscenes_vehicle', 'data.dataset_dir=/media/datasets/nuscenes', 'data.labels_dir=/media/datasets/cvt_labels_nuscenes']
Traceback (most recent call last):
File "scripts/train.py", line 71, in main
trainer.fit(model_module, datamodule=data_module, ckpt_path=ckpt_path)
(traceback to pytorch stuff)
ckpt_path=ckpt_path)pytorch_lightning.utilities.exceptions.MisconfigurationException: The provided lr scheduler OneCycleLR doesn't follow PyTorch's LRScheduler API. You should override the LightningModule.lr_scheduler_step hook with your own logic if you are using a custom LR scheduler.