Multi gpu training about nas_fpn_tensorflow HOT 6 CLOSED

can you show your cfgs.py? @unlabeledData

from nas_fpn_tensorflow.

OK, I changed some lines in cfgs.py.

GPU_GROUP = "1,2,3,4"
VERSION = 'FPN_Res50_COCO_20190503_v5'
NET_NAME = 'resnet50_v1d'
#NET_NAME = 'resnet_v1_50'
#NET_NAME = 'MobilenetV2'
ADD_BOX_IN_TENSORBOARD = True

#------------------------------------------ Train config
CUDA9 = False

# -------------------------------------------- Data_preprocess_config
# DATASET_NAME = 'coco'
# PIXEL_MEAN = [123.68, 116.779, 103.939] # R, G, B. In tf, channel is RGB. In openCV, channel is BGR
# PIXEL_MEAN_ = [0.485, 0.456, 0.406]
# PIXEL_STD = [0.229, 0.224, 0.225]
# IMG_SHORT_SIDE_LEN = 800
# IMG_MAX_LENGTH = 1333
# CLASS_NUM = 80
DATASET_NAME = 'pascal'
PIXEL_MEAN = [123.68, 116.779, 103.939] # R, G, B. In tf, channel is RGB. In openCV, channel is BGR
PIXEL_MEAN_ = [0.485, 0.456, 0.406]
PIXEL_STD = [0.229, 0.224, 0.225]
IMG_SHORT_SIDE_LEN = 800
IMG_MAX_LENGTH = 1333
CLASS_NUM = 20

Others is the same as your cfgs.py.

from nas_fpn_tensorflow.

/home/DATA3/sunli-work/NAS_FPN_Tensorflow/data/pretrained_weights/resnet50_v1d.ckpt
do you put weights here? @unlabeledData

from nas_fpn_tensorflow.

Yes, I put resnet50_v1d.ckpt there.

from nas_fpn_tensorflow.

have you resolve this problem？ @unlabeledData

from nas_fpn_tensorflow.

have you resolve this problem？ @unlabeledData

Not yet. I didn't use this project.
Maybe you can try anther res50 pretraining models.
The authors' FPN_Tensorflow worked well.

I want to close this issue after you see the comment.

from nas_fpn_tensorflow.