The if condition is not sufficient
when you trian the det model , there is not the var: loss_type in the Global, so train_alg_type == 'rec' is necessary
This commit is contained in:
parent
eca0ef34ec
commit
dcd2626e11
|
@ -74,7 +74,7 @@ def main():
|
|||
|
||||
# dump mode structure
|
||||
if config['Global']['debug']:
|
||||
if 'attention' in config['Global']['loss_type']:
|
||||
if train_alg_type == 'rec' and 'attention' in config['Global']['loss_type']:
|
||||
logger.warning('Does not suport dump attention...')
|
||||
else:
|
||||
summary(train_program)
|
||||
|
|
Loading…
Reference in New Issue