├── .gitignore ├── LICENSE ├── README.md ├── dawn ├── logs │ └── single_machine │ │ ├── debug.log │ │ ├── event.log │ │ ├── events.out.tfevents.1536151072.ip-192-168-84-137 │ │ └── verbose.log ├── prepare_dawn_bs.py ├── prepare_dawn_is.py ├── prepare_dawn_lr.py └── prepare_dawn_tsv.py ├── requirements.txt ├── setup.sh ├── tensorboard.png ├── tools ├── create_imagenet_snapshot.py ├── launch_tensorboard.py └── replicate_imagenet.py ├── train.py ├── training ├── dataloader.py ├── dist_utils.py ├── experimental_utils.py ├── fp16util.py ├── logger.py ├── meter.py ├── resnet.py └── train_imagenet_nv.py ├── util.py └── worker_requirements.txt /.gitignore: -------------------------------------------------------------------------------- 1 | *.DS_Store 2 | __pycache__ 3 | /.idea 4 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/README.md -------------------------------------------------------------------------------- /dawn/logs/single_machine/debug.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/dawn/logs/single_machine/debug.log -------------------------------------------------------------------------------- /dawn/logs/single_machine/event.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/dawn/logs/single_machine/event.log -------------------------------------------------------------------------------- /dawn/logs/single_machine/events.out.tfevents.1536151072.ip-192-168-84-137: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/dawn/logs/single_machine/events.out.tfevents.1536151072.ip-192-168-84-137 -------------------------------------------------------------------------------- /dawn/logs/single_machine/verbose.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/dawn/logs/single_machine/verbose.log -------------------------------------------------------------------------------- /dawn/prepare_dawn_bs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/dawn/prepare_dawn_bs.py -------------------------------------------------------------------------------- /dawn/prepare_dawn_is.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/dawn/prepare_dawn_is.py -------------------------------------------------------------------------------- /dawn/prepare_dawn_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/dawn/prepare_dawn_lr.py -------------------------------------------------------------------------------- /dawn/prepare_dawn_tsv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/dawn/prepare_dawn_tsv.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/setup.sh -------------------------------------------------------------------------------- /tensorboard.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/tensorboard.png -------------------------------------------------------------------------------- /tools/create_imagenet_snapshot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/tools/create_imagenet_snapshot.py -------------------------------------------------------------------------------- /tools/launch_tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/tools/launch_tensorboard.py -------------------------------------------------------------------------------- /tools/replicate_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/tools/replicate_imagenet.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/train.py -------------------------------------------------------------------------------- /training/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/training/dataloader.py -------------------------------------------------------------------------------- /training/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/training/dist_utils.py -------------------------------------------------------------------------------- /training/experimental_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/training/experimental_utils.py -------------------------------------------------------------------------------- /training/fp16util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/training/fp16util.py -------------------------------------------------------------------------------- /training/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/training/logger.py -------------------------------------------------------------------------------- /training/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/training/meter.py -------------------------------------------------------------------------------- /training/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/training/resnet.py -------------------------------------------------------------------------------- /training/train_imagenet_nv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/training/train_imagenet_nv.py -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/util.py -------------------------------------------------------------------------------- /worker_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cybertronai/imagenet18/HEAD/worker_requirements.txt --------------------------------------------------------------------------------