we visualize our training details via wandb (https://wandb.ai/site).
- you'll need to login
$ wandb login
- you'll need to copy & paste you API key in terminal
or add the key to the "code/config/config.py" with
$ https://wandb.ai/authorize
C.wandb_key = ""
our code is trained with 4 x nvidia a100 gpus, alternatively, you can downsize the range view's input resolution to lower the training requirements, but please note that the final results cannot be guaranteed to match ours.
for training, please find the training scripts in "scripts" directory.
$ ./scripts/nuscenes_run.sh your_defined_labelled_num
$ ./scripts/kitti_run.sh your_defined_labelled_num
checkpoints and training logs are in this google drive link.
some examples of training detail, please see this wandb link.
In details, after clicking the run (e.g., nuscenes_final_10_pct), you can checkout: