MCPcopy Index your code
hub / github.com/tensorpack/tensorpack / get_config

Function get_config

examples/Char-RNN/char-rnn.py:111–125  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

109
110
111def get_config():
112 logger.auto_set_dir()
113
114 ds = CharRNNData(param.corpus, 100000)
115 ds = BatchData(ds, param.batch_size)
116
117 return TrainConfig(
118 data=QueueInput(ds),
119 callbacks=[
120 ModelSaver(),
121 ScheduledHyperParamSetter('learning_rate', [(25, 2e-4)])
122 ],
123 model=Model(),
124 max_epoch=50,
125 )
126
127
128def sample(path, start, length):

Callers 1

char-rnn.pyFile · 0.70

Calls 7

CharRNNDataClass · 0.85
BatchDataClass · 0.85
TrainConfigClass · 0.85
QueueInputClass · 0.85
ModelSaverClass · 0.85
ModelClass · 0.70

Tested by

no test coverage detected