├── .gitignore ├── LICENSE ├── NOTICE ├── README.md ├── augment ├── __init__.py ├── gaussian_blur.py ├── getters.py ├── normalize.py └── transforms.py ├── config ├── byol_1000ep.yaml ├── byol_200ep.yaml ├── scrl_1000ep.yaml └── scrl_200ep.yaml ├── data ├── __init__.py ├── dataloader.py ├── datasets.py └── getters.py ├── distributed ├── __init__.py ├── comm.py ├── getters.py └── launch.py ├── images └── overview.png ├── main.py ├── models ├── __init__.py ├── backbone.py ├── box_generator.py └── heads.py ├── optim ├── __init__.py ├── getters.py ├── optim.py └── scheduler.py ├── requirements.txt ├── trainer ├── __init__.py ├── base.py ├── helper.py ├── linear_eval.py ├── result_pack.py └── trainer.py └── utils ├── __init__.py ├── color.py ├── config.py ├── exception_logger.py ├── logger.py ├── parser.py ├── progress_disp.py ├── stderr_redirector.py ├── utils.py └── watch.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/README.md -------------------------------------------------------------------------------- /augment/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/augment/__init__.py -------------------------------------------------------------------------------- /augment/gaussian_blur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/augment/gaussian_blur.py -------------------------------------------------------------------------------- /augment/getters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/augment/getters.py -------------------------------------------------------------------------------- /augment/normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/augment/normalize.py -------------------------------------------------------------------------------- /augment/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/augment/transforms.py -------------------------------------------------------------------------------- /config/byol_1000ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/config/byol_1000ep.yaml -------------------------------------------------------------------------------- /config/byol_200ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/config/byol_200ep.yaml -------------------------------------------------------------------------------- /config/scrl_1000ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/config/scrl_1000ep.yaml -------------------------------------------------------------------------------- /config/scrl_200ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/config/scrl_200ep.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/data/dataloader.py -------------------------------------------------------------------------------- /data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/data/datasets.py -------------------------------------------------------------------------------- /data/getters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/data/getters.py -------------------------------------------------------------------------------- /distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/distributed/__init__.py -------------------------------------------------------------------------------- /distributed/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/distributed/comm.py -------------------------------------------------------------------------------- /distributed/getters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/distributed/getters.py -------------------------------------------------------------------------------- /distributed/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/distributed/launch.py -------------------------------------------------------------------------------- /images/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/images/overview.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/models/backbone.py -------------------------------------------------------------------------------- /models/box_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/models/box_generator.py -------------------------------------------------------------------------------- /models/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/models/heads.py -------------------------------------------------------------------------------- /optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/optim/__init__.py -------------------------------------------------------------------------------- /optim/getters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/optim/getters.py -------------------------------------------------------------------------------- /optim/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/optim/optim.py -------------------------------------------------------------------------------- /optim/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/optim/scheduler.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/requirements.txt -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/trainer/__init__.py -------------------------------------------------------------------------------- /trainer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/trainer/base.py -------------------------------------------------------------------------------- /trainer/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/trainer/helper.py -------------------------------------------------------------------------------- /trainer/linear_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/trainer/linear_eval.py -------------------------------------------------------------------------------- /trainer/result_pack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/trainer/result_pack.py -------------------------------------------------------------------------------- /trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/trainer/trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/color.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/utils/color.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/exception_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/utils/exception_logger.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/utils/parser.py -------------------------------------------------------------------------------- /utils/progress_disp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/utils/progress_disp.py -------------------------------------------------------------------------------- /utils/stderr_redirector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/utils/stderr_redirector.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/watch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kakaobrain/scrl/HEAD/utils/watch.py --------------------------------------------------------------------------------