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'