czzhangheng
/
DCRNN
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updated epeoch num

This commit is contained in:
Chintan Shah 2019-10-08 17:32:16 -04:00
parent b2d2b21dbd
commit 073f1d4a6e
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
data/model/dcrnn_test_config.yaml
View File

@ -25,7 +25,7 @@ model:
train: train:
base_lr: 0.01 base_lr: 0.01
dropout: 0 dropout: 0
epoch: 44 epoch: 51
epochs: 100 epochs: 100
epsilon: 1.0e-3 epsilon: 1.0e-3
global_step: 0 global_step: 0

2
run_demo.py
View File

@ -32,6 +32,6 @@ if __name__ == '__main__':
parser.add_argument('--use_cpu_only', default=False, type=str, help='Whether to run tensorflow on cpu.') parser.add_argument('--use_cpu_only', default=False, type=str, help='Whether to run tensorflow on cpu.')
parser.add_argument('--config_filename', default='data/model/pretrained/METR-LA/config.yaml', type=str, parser.add_argument('--config_filename', default='data/model/pretrained/METR-LA/config.yaml', type=str,
help='Config file for pretrained model.') help='Config file for pretrained model.')
parser.add_argument('--output_filename', default='data/dcrnn_predictions.npz') parser.add_argument('--output_filename', default='data/dcrnn_predictions_tf.npz')
args = parser.parse_args() args = parser.parse_args()
run_dcrnn(args) run_dcrnn(args)