fix reload.

lib/model/config.py

@@ -269,9 +269,6 @@
 # Use GPU implementation of non-maximum suppression
 __C.USE_GPU_NMS = True
 
-# Default GPU device id
-__C.GPU_ID = 0
-
 # Default pooling mode, only 'crop' is available
 __C.POOLING_MODE = 'crop'
 

lib/model/nms_wrapper.py

@@ -18,6 +18,6 @@ def nms(dets, thresh, force_cpu=False):
   if dets.shape[0] == 0:
     return []
   if cfg.USE_GPU_NMS and not force_cpu:
-    return gpu_nms(dets, thresh, device_id=cfg.GPU_ID)
+    return gpu_nms(dets, thresh, device_id=0)
   else:
     return cpu_nms(dets, thresh)

lib/model/train_val.py

@@ -80,7 +80,7 @@ def snapshot(self, sess, iter):
     return filename, nfilename
 
   def from_snapshot(self, sess, sfile, nfile):
-    print('Restoring model snapshots from {:s}'.format(sfiles[-1]))
+    print('Restoring model snapshots from {:s}'.format(sfile))
     self.saver.restore(sess, sfile)
     print('Restored.')
     # Needs to restore the other hyper-parameters/states for training, (TODO xinlei) I have
@@ -363,7 +363,7 @@ def is_valid(entry):
 def train_net(network, imdb, roidb, valroidb, output_dir, tb_dir,
               pretrained_model=None,
               max_iters=40000):
-  """Train a Fast R-CNN network."""
+  """Train a Faster R-CNN network."""
   roidb = filter_roidb(roidb)
   valroidb = filter_roidb(valroidb)