(name='v0', hparams=None)
| 495 | |
| 496 | |
| 497 | def auto_augment_policy(name='v0', hparams=None): |
| 498 | hparams = hparams or _HPARAMS_DEFAULT |
| 499 | if name == 'original': |
| 500 | return auto_augment_policy_original(hparams) |
| 501 | elif name == 'originalr': |
| 502 | return auto_augment_policy_originalr(hparams) |
| 503 | elif name == 'v0': |
| 504 | return auto_augment_policy_v0(hparams) |
| 505 | elif name == 'v0r': |
| 506 | return auto_augment_policy_v0r(hparams) |
| 507 | else: |
| 508 | assert False, 'Unknown AA policy (%s)' % name |
| 509 | |
| 510 | |
| 511 | class AutoAugment: |
no test coverage detected