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:
base_lr: 0.01
dropout: 0
epoch: 44
epoch: 51
epochs: 100
epsilon: 1.0e-3
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('--config_filename', default='data/model/pretrained/METR-LA/config.yaml', type=str,
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()
run_dcrnn(args)