图像分类开启显存优化会hang住
Created by: ccmeteorljh
FLAGS_fraction_of_gpu_memory_to_use=0.0 CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 sh run.sh
----------- Configuration Arguments -----------
batch_size: 256
checkpoint: None
class_dim: 1000
data_dir: ./data/ILSVRC2012
enable_ce: False
fp16: False
image_shape: 3,224,224
l2_decay: 0.0001
lr: 0.1
lr_strategy: piecewise_decay
model: SE_ResNeXt50_32x4d
model_category: models_name
model_save_dir: output/
momentum_rate: 0.9
num_epochs: 120
pretrained_model: None
scale_loss: 1.0
total_images: 1281167
use_gpu: True
with_mem_opt: 1
------------------------------------------------
W0311 06:36:14.564357 3503 device_context.cc:263] Please NOTE: device: 0, CUDA Capability: 70, Driver API Version: 9.0, Runtime API Version: 9.0
W0311 06:36:14.564435 3503 device_context.cc:271] device: 0, cuDNN Version: 7.0.
W0311 06:36:14.564447 3503 device_context.cc:295] WARNING: device: 0. The installed Paddle is compiled with CUDNN 7.3, but CUDNN version in your machine is 7.0, which may cause serious incompatible bug. Please recompile or reinstall Paddle with compatible CUDNN version.
Pass 0, trainbatch 0, loss 6.92884540558, acc1 0.00390625, acc5 0.0078125, lr0.10000, time 7.06 sec
/usr/local/lib/python2.7/dist-packages/PIL/TiffImagePlugin.py:763: UserWarning: Possibly corrupt EXIF data. Expecting to read 1835008 bytes but only got 0. Skipping tag 0
" Skipping tag %s" % (size, len(data), tag))
/usr/local/lib/python2.7/dist-packages/PIL/TiffImagePlugin.py:763: UserWarning: Possibly corrupt EXIF data. Expecting to read 2555904 bytes but only got 0. Skipping tag 0
" Skipping tag %s" % (size, len(data), tag))