DeepLab V3+：ResourceExhaustedError (see above for traceback): OOM when allocating tensor of shape

在跑DeepLab V3+时遇到ResourceExhaustedError (see above for traceback): OOM when allocating tensor of shape错误

GPU：1060/6G

github：https://github.com/tensorflow/models/tree/master/research/deeplab

是显存不够了，把local_test.sh文件中

python "${WORK_DIR}"/train.py

--logtostderr

--train_split="trainval"

--model_variant="xception_65"

--atrous_rates=6

--atrous_rates=12

--atrous_rates=18

--output_stride=16

--decoder_output_stride=4

--train_crop_size=513

--train_crop_size=513

--train_batch_size=4

--training_number_of_steps="${NUM_ITERATIONS}"

--fine_tune_batch_norm=true   <==============改这行

--tf_initial_checkpoint="${INIT_FOLDER}/deeplabv3_pascal_train_aug/model.ckpt"

--train_logdir="${TRAIN_LOGDIR}"

--dataset_dir="${PASCAL_DATASET}"

改为

python "${WORK_DIR}"/train.py

--logtostderr

--train_split="trainval"

--model_variant="xception_65"

--atrous_rates=6

--atrous_rates=12

--atrous_rates=18

--output_stride=16

--decoder_output_stride=4

--train_crop_size=513

--train_crop_size=513

--train_batch_size=1

--training_number_of_steps="${NUM_ITERATIONS}"

--fine_tune_batch_norm=false

--tf_initial_checkpoint="${INIT_FOLDER}/deeplabv3_pascal_train_aug/model.ckpt"

--train_logdir="${TRAIN_LOGDIR}"

--dataset_dir="${PASCAL_DATASET}"

之后正常训练

最后

以上就是自然黑夜最近收集整理的关于DeepLab V3+：ResourceExhaustedError (see above for traceback): OOM when allocating tensor of shape的全部内容，更多相关DeepLab内容请搜索靠谱客的其他文章。