├── .github └── workflows │ ├── docs.yml │ ├── linter.yaml │ └── tests.yaml ├── .gitignore ├── CHANGELOG.md ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── Makefile ├── README.md ├── examples ├── basic │ ├── __init__.py │ ├── config │ │ └── config.yaml │ └── train.py └── cifar │ ├── __init__.py │ ├── config │ └── config.yaml │ ├── solver.py │ └── train.py ├── flashy ├── __init__.py ├── adversarial.py ├── distrib.py ├── formatter.py ├── loggers │ ├── __init__.py │ ├── base.py │ ├── localfs.py │ ├── tensorboard.py │ ├── utils.py │ └── wandb.py ├── logging.py ├── py.typed ├── solver.py ├── state.py └── utils.py ├── mypy.ini ├── pyproject.toml ├── requirements.txt ├── setup.cfg ├── setup.py └── tests ├── __init__.py ├── dummy ├── __init__.py ├── conf │ └── config.yaml └── train.py ├── test_distrib.py ├── test_formatter.py ├── test_integ.py ├── test_solver.py └── test_state.py /.github/workflows/docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/.github/workflows/docs.yml -------------------------------------------------------------------------------- /.github/workflows/linter.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/.github/workflows/linter.yaml -------------------------------------------------------------------------------- /.github/workflows/tests.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/.github/workflows/tests.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/README.md -------------------------------------------------------------------------------- /examples/basic/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/examples/basic/__init__.py -------------------------------------------------------------------------------- /examples/basic/config/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/examples/basic/config/config.yaml -------------------------------------------------------------------------------- /examples/basic/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/examples/basic/train.py -------------------------------------------------------------------------------- /examples/cifar/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/examples/cifar/__init__.py -------------------------------------------------------------------------------- /examples/cifar/config/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/examples/cifar/config/config.yaml -------------------------------------------------------------------------------- /examples/cifar/solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/examples/cifar/solver.py -------------------------------------------------------------------------------- /examples/cifar/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/examples/cifar/train.py -------------------------------------------------------------------------------- /flashy/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/__init__.py -------------------------------------------------------------------------------- /flashy/adversarial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/adversarial.py -------------------------------------------------------------------------------- /flashy/distrib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/distrib.py -------------------------------------------------------------------------------- /flashy/formatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/formatter.py -------------------------------------------------------------------------------- /flashy/loggers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/loggers/__init__.py -------------------------------------------------------------------------------- /flashy/loggers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/loggers/base.py -------------------------------------------------------------------------------- /flashy/loggers/localfs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/loggers/localfs.py -------------------------------------------------------------------------------- /flashy/loggers/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/loggers/tensorboard.py -------------------------------------------------------------------------------- /flashy/loggers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/loggers/utils.py -------------------------------------------------------------------------------- /flashy/loggers/wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/loggers/wandb.py -------------------------------------------------------------------------------- /flashy/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/logging.py -------------------------------------------------------------------------------- /flashy/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /flashy/solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/solver.py -------------------------------------------------------------------------------- /flashy/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/state.py -------------------------------------------------------------------------------- /flashy/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/flashy/utils.py -------------------------------------------------------------------------------- /mypy.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/mypy.ini -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- 1 | [build-system] 2 | requires = ["setuptools", "wheel"] 3 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | dora_search 2 | colorlog 3 | numpy 4 | torch 5 | -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/dummy/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/tests/dummy/__init__.py -------------------------------------------------------------------------------- /tests/dummy/conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/tests/dummy/conf/config.yaml -------------------------------------------------------------------------------- /tests/dummy/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/tests/dummy/train.py -------------------------------------------------------------------------------- /tests/test_distrib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/tests/test_distrib.py -------------------------------------------------------------------------------- /tests/test_formatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/tests/test_formatter.py -------------------------------------------------------------------------------- /tests/test_integ.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/tests/test_integ.py -------------------------------------------------------------------------------- /tests/test_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/tests/test_solver.py -------------------------------------------------------------------------------- /tests/test_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/flashy/HEAD/tests/test_state.py --------------------------------------------------------------------------------