├── .gitignore ├── README.md ├── agent ├── __init__.py ├── base.py ├── ppo.py └── reinforce.py ├── aug_op ├── __init__.py ├── ops.py └── registry.py ├── criterion ├── __init__.py └── labelsmooth.py ├── distm ├── __init__.py ├── base.py ├── local.py └── torch.py ├── main.py ├── models ├── __init__.py ├── resnet.py └── wresnet.py ├── pipeline ├── __init__.py ├── aws.py ├── base.py └── ohl.py ├── requirements.txt ├── scheduler ├── __init__.py └── scheduler.py └── utils ├── config.py ├── data.py ├── dist.py ├── file.py └── misc.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/README.md -------------------------------------------------------------------------------- /agent/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/agent/__init__.py -------------------------------------------------------------------------------- /agent/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/agent/base.py -------------------------------------------------------------------------------- /agent/ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/agent/ppo.py -------------------------------------------------------------------------------- /agent/reinforce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/agent/reinforce.py -------------------------------------------------------------------------------- /aug_op/__init__.py: -------------------------------------------------------------------------------- 1 | from .ops import aug_ops_dict 2 | -------------------------------------------------------------------------------- /aug_op/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/aug_op/ops.py -------------------------------------------------------------------------------- /aug_op/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/aug_op/registry.py -------------------------------------------------------------------------------- /criterion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/criterion/__init__.py -------------------------------------------------------------------------------- /criterion/labelsmooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/criterion/labelsmooth.py -------------------------------------------------------------------------------- /distm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/distm/__init__.py -------------------------------------------------------------------------------- /distm/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/distm/base.py -------------------------------------------------------------------------------- /distm/local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/distm/local.py -------------------------------------------------------------------------------- /distm/torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/distm/torch.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/wresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/models/wresnet.py -------------------------------------------------------------------------------- /pipeline/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/pipeline/__init__.py -------------------------------------------------------------------------------- /pipeline/aws.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/pipeline/aws.py -------------------------------------------------------------------------------- /pipeline/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/pipeline/base.py -------------------------------------------------------------------------------- /pipeline/ohl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/pipeline/ohl.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/requirements.txt -------------------------------------------------------------------------------- /scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/scheduler/__init__.py -------------------------------------------------------------------------------- /scheduler/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/scheduler/scheduler.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/utils/data.py -------------------------------------------------------------------------------- /utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/utils/dist.py -------------------------------------------------------------------------------- /utils/file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/utils/file.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Awesome-AutoAug-Algorithms/AWS-OHL-AutoAug/HEAD/utils/misc.py --------------------------------------------------------------------------------