Wei-Ching Wang

training on coco dataset

1

has anyone ran into issue training on CocoDataset , I have modified on line 220: elif args.dataset_type == 'coco': 221 # Image preprocessing, normalization for the pretrained resnet 222 transform...

running train.py as followed encounter the error (pytorch_p27) [ec2-user@ip-172-31-37-28 SSD-variants]$ python train.py --demo --voc_root /VOCdevkit --checkpoint vgg16_reducedfc.pth File "train.py", line 27, in from tensorboardX import SummaryWriter File "/home/ec2-user/anaconda3/envs/pytorch_p27/lib/python2.7/site-packages/tensorboardX/__init__.py", line 5,...

tf_container.py modification for running GPU context

2

would like to start GPU session in container, and tried the following: if (len(frozen_graph_exists) == 0): with tf.Graph().as_default() as graph: self.sess = tf.Session(graph=graph) loader.load(self.sess, [tf.saved_model.tag_constants.SERVING], os.path.join(path, "tfmodel")) else: self.sess =...