IRMVLab / EfficientLO-Net

EfficientLO-Net: Efficient 3D Deep LiDAR Odometry (PAMI 2022)
68 stars 4 forks source link

Train Error TypeError: train_one_epoch() got an unexpected keyword argument 'train_list' #7

Open wennycooper opened 1 day ago

wennycooper commented 1 day ago

Hi, I tried to run command_train.sh but I got following error. Can anybody help?

python main.py \
    --mode train \
    --gpu 1 \
    --model pwclo_model \
    --data_root /mnt/SardinaDatasets/kitti_odom/dataset/sequences \
    --checkpoint_path ./pretrained_model/pretrained_model.ckpt \
    --log_dir train_log_ \
    --result_dir result \
    --train_list 0 1 2 3 4 5 6 \
    --val_list 7 8 9 10 \
    --test_list 0 1 2 3 4 5 6 7 8 9 10 \
    --num_H_input 64 \
    --num_W_input 1800 \
    --max_epoch 1000 \
    --learning_rate 0.001 \
    --batch_size 8 \
    > Train_Efficient-LOnet_log.txt 2>&1 &
pid: 1334
--- Get model and loss
--- Get training operator
model restored
**** EPOCH 000 ****
Traceback (most recent call last):
  File "main.py", line 605, in <module>
    main(MODE)
  File "main.py", line 224, in main
    train_one_epoch(sess, ops, train_writer, train_list = TRAIN_LIST)
TypeError: train_one_epoch() got an unexpected keyword argument 'train_list'