报错“ValueError: invalid literal for int() with base 10: './saved_model/unet_pet/10”
Created by: lakebodhi
根据https://github.com/PaddlePaddle/PaddleSeg/blob/release/v0.2.0/docs/usage.md的示例操作,执行
python pdseg/train.py --do_eval \ --use_tb \ --tb_log_dir train_log \ --cfg configs/unet_pet.yaml \ BATCH_SIZE 4 \ TRAIN.PRETRAINED_MODEL_DIR pretrained_model/unet_bn_coco \ SOLVER.LR 5e-5
程序报错:
#1 visualize image path: visual\visual_results\Abyssinian_201.png Traceback (most recent call last): File "pdseg/train.py", line 487, in main(args) File "pdseg/train.py", line 474, in main train(cfg) File "pdseg/train.py", line 456, in train log_writer=log_writer) File "D:\remote_sensing\PaddleSeg\pdseg\vis.py", line 229, in visualize epoch = int(ckpt_dir.split(os.path.sep)[-1]) ValueError: invalid literal for int() with base 10: './saved_model/unet_pet/10'
操作系统windows10 64位 cpu训练