()
| 109 | |
| 110 | |
| 111 | def 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 | |
| 128 | def sample(path, start, length): |
no test coverage detected