Skip to content

Commit 50673bd

Browse files
committed
PyTorch demo (wip)
#1120
1 parent 6051f8a commit 50673bd

File tree

1 file changed

+38
-0
lines changed

1 file changed

+38
-0
lines changed

demos/demo-torch.config

+38
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
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

0 commit comments

Comments
 (0)