Syed Abdul Gaffar Shakhadri commited on
Commit
2826c26
·
unverified ·
1 Parent(s): 024304f

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +2 -2
config.py CHANGED
@@ -63,7 +63,7 @@ _C.MODEL.NAME = ''
63
  # Checkpoint to resume, could be overwritten by command line argument
64
  _C.MODEL.RESUME = ''
65
  # Number of classes, overwritten in data preparation
66
- _C.MODEL.NUM_CLASSES = 200#1000
67
  # Dropout rate
68
  _C.MODEL.DROP_RATE = 0.0
69
  # Drop path rate
@@ -291,4 +291,4 @@ def get_inference_config(cfg_path):
291
 
292
  return config
293
 
294
- ################### For Inferencing ####################
 
63
  # Checkpoint to resume, could be overwritten by command line argument
64
  _C.MODEL.RESUME = ''
65
  # Number of classes, overwritten in data preparation
66
+ _C.MODEL.NUM_CLASSES = 1000
67
  # Dropout rate
68
  _C.MODEL.DROP_RATE = 0.0
69
  # Drop path rate
 
291
 
292
  return config
293
 
294
+ ################### For Inferencing ####################