Get architecture name.
(output_dir, model_arch)
| 1014 | |
| 1015 | |
| 1016 | def get_checkpoint_dir(output_dir, model_arch): |
| 1017 | """Get architecture name.""" |
| 1018 | arch_name = model_arch.__name__ |
| 1019 | return os.path.join(output_dir, arch_name) |
| 1020 | |
| 1021 | |
| 1022 | def build_model(model_arch, tiny_config, output_dir, keep_model=False): |
no test coverage detected