Diff of /MainDecExample.py [000000] .. [6d7935]

Switch to unified view

a b/MainDecExample.py
1
"""
2
3
Stefania Fresca, MOX Laboratory, Politecnico di Milano
4
April 2019
5
6
"""
7
8
import os
9
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
10
import sys
11
sys.stdout = open('*.out', 'w')
12
13
import utils
14
from DecNet import DecNet
15
16
if __name__ == '__main__':
17
    config = dict()
18
    config['n'] =
19
    config['n_params'] =
20
    config['lr'] =
21
    config['omega_h'] =
22
    config['omega_n'] =
23
    config['batch_size'] =
24
    config['n_data'] =
25
    config['N_h'] =
26
    config['n_h'] =
27
    config['N_t'] =
28
    config['train_mat'] = ''
29
    config['test_mat'] = ''
30
    config['train_params'] = ''
31
    config['test_params'] = ''
32
    config['checkpoints_folder'] = ''
33
    config['graph_folder'] = ''
34
    config['large'] =
35
    config['zero_padding'] =
36
    config['p'] =
37
    config['restart'] =
38
39
    model = DecNet(config)
40
    model.build()
41
    model.test_all()