File tree 1 file changed +38
-0
lines changed
1 file changed +38
-0
lines changed Original file line number Diff line number Diff line change
1
+ #!returnn.py
2
+ # kate: syntax python;
3
+ # -*- mode: python -*-
4
+ # sublime: syntax 'Packages/Python Improved/PythonImproved.tmLanguage'
5
+ # vim:set expandtab tabstop=4 fenc=utf-8 ff=unix ft=python:
6
+
7
+ import os
8
+ from returnn .util .basic import get_login_username
9
+
10
+ demo_name , _ = os .path .splitext (__file__ )
11
+ print ("Hello, experiment: %s" % demo_name )
12
+
13
+ backend = "torch"
14
+
15
+ task = "train"
16
+ train = {"class" : "Task12AXDataset" , "num_seqs" : 1000 }
17
+ dev = {"class" : "Task12AXDataset" , "num_seqs" : 100 , "fixed_random_seed" : 1 }
18
+
19
+ num_inputs = 9
20
+ num_outputs = 2
21
+ batching = "random"
22
+ batch_size = 5000
23
+ max_seqs = 10
24
+ chunking = "200:200"
25
+
26
+ # TODO define network and loss...
27
+
28
+ # training
29
+ # TODO maybe remove this...
30
+ optimizer = {"class" : "adam" }
31
+ learning_rate = 0.01
32
+
33
+ model = "/tmp/%s/returnn/%s/model" % (get_login_username (), demo_name )
34
+ num_epochs = 5
35
+
36
+ # log
37
+ #log_verbosity = 3
38
+ log_verbosity = 5
You can’t perform that action at this time.
0 commit comments