()
| 73 | |
| 74 | |
| 75 | def get_config(): |
| 76 | dataset_train = get_data('train') |
| 77 | dataset_val = get_data('val') |
| 78 | |
| 79 | return TrainConfig( |
| 80 | model=Model(), |
| 81 | dataflow=dataset_train, |
| 82 | callbacks=[ |
| 83 | ModelSaver(), |
| 84 | PeriodicTrigger(InferenceRunner(dataset_val, [ |
| 85 | ClassificationError('wrong-top1', 'val-error-top1'), |
| 86 | ClassificationError('wrong-top5', 'val-error-top5')]), |
| 87 | every_k_epochs=2), |
| 88 | ScheduledHyperParamSetter('learning_rate', |
| 89 | [(30, 1e-2), (55, 1e-3), (75, 1e-4), (95, 1e-5)]), |
| 90 | ], |
| 91 | steps_per_epoch=5000, |
| 92 | max_epoch=105, |
| 93 | ) |
| 94 | |
| 95 | |
| 96 | def viz_cam(model_file, data_dir): |
no test coverage detected
searching dependent graphs…