Syed Abdul Gaffar Shakhadri
commited on
Update config.py
Browse files
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 =
|
| 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 ####################
|