update kes_model.py
This commit is contained in:
parent
fb10439f2f
commit
215cff687d
@ -122,7 +122,6 @@ def init_model(configs):
|
|||||||
else:
|
else:
|
||||||
print('Unknown body type {}'.format(backbone_type))
|
print('Unknown body type {}'.format(backbone_type))
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
if 'classifier' in configs:
|
if 'classifier' in configs:
|
||||||
classifier_type = configs['classifier']['type']
|
classifier_type = configs['classifier']['type']
|
||||||
dropout = configs['classifier']['dropout']
|
dropout = configs['classifier']['dropout']
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user