Spaces:
Running
Running
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 ####################
|