Skip to content

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

@wennycooper

Description

@wennycooper

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'

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions