Skip to content

Commit a6b7a22

Browse files
Merge pull request #406 from HydrogenSulfate/fix_attentionLSTM_cfg
fix attentionLSTM tipc's config file
2 parents 941ed9a + b12db2e commit a6b7a22

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

test_tipc/configs/AttentionLSTM/train_amp_infer_python.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,13 +21,13 @@ amp_train:main.py --amp --amp_level='O2' --validate -c configs/recognition/atten
2121
null:null
2222
##
2323
===========================eval_params===========================
24-
eval:main.py --test -c configs/recognition/attention_lstm/attention_lstm_youtube8m.yaml --save_name inference
24+
eval:main.py --test -c configs/recognition/attention_lstm/attention_lstm_youtube8m.yaml
2525
-w:./test_tipc/output/AttentionLSTM/AttentionLSTM_epoch_00001.pdparams
2626
##
2727
===========================infer_params===========================
2828
-o:inference/AttentionLSTM
2929
-p:null
30-
norm_export:tools/export_model.py -c configs/recognition/attention_lstm/attention_lstm_youtube8m.yaml
30+
norm_export:tools/export_model.py -c configs/recognition/attention_lstm/attention_lstm_youtube8m.yaml --save_name inference
3131
quant_export:null
3232
fpgm_export:null
3333
distill_export:null

test_tipc/configs/AttentionLSTM/train_infer_python.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,13 +21,13 @@ null:null
2121
null:null
2222
##
2323
===========================eval_params===========================
24-
eval:main.py --test -c configs/recognition/attention_lstm/attention_lstm_youtube8m.yaml --save_name inference
24+
eval:main.py --test -c configs/recognition/attention_lstm/attention_lstm_youtube8m.yaml
2525
-w:./test_tipc/output/AttentionLSTM/AttentionLSTM_epoch_00001.pdparams
2626
##
2727
===========================infer_params===========================
2828
-o:inference/AttentionLSTM
2929
-p:null
30-
norm_export:tools/export_model.py -c configs/recognition/attention_lstm/attention_lstm_youtube8m.yaml
30+
norm_export:tools/export_model.py -c configs/recognition/attention_lstm/attention_lstm_youtube8m.yaml --save_name inference
3131
quant_export:null
3232
fpgm_export:null
3333
distill_export:null

0 commit comments

Comments
 (0)