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
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions