{"id":21389479,"url":"https://github.com/feifeibear/horovod-resnet","last_synced_at":"2025-10-15T04:36:44.427Z","repository":{"id":85016982,"uuid":"122920350","full_name":"feifeibear/horovod-resnet","owner":"feifeibear","description":null,"archived":false,"fork":false,"pushed_at":"2018-03-04T23:31:20.000Z","size":28,"stargazers_count":4,"open_issues_count":0,"forks_count":0,"subscribers_count":3,"default_branch":"master","last_synced_at":"2025-01-23T00:41:16.285Z","etag":null,"topics":[],"latest_commit_sha":null,"homepage":null,"language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":null,"status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/feifeibear.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":null,"code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2018-02-26T05:55:14.000Z","updated_at":"2024-04-29T14:31:20.000Z","dependencies_parsed_at":"2023-05-30T09:45:55.695Z","dependency_job_id":null,"html_url":"https://github.com/feifeibear/horovod-resnet","commit_stats":null,"previous_names":[],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/feifeibear%2Fhorovod-resnet","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/feifeibear%2Fhorovod-resnet/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/feifeibear%2Fhorovod-resnet/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/feifeibear%2Fhorovod-resnet/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/feifeibear","download_url":"https://codeload.github.com/feifeibear/horovod-resnet/tar.gz/refs/heads/master","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":243871652,"owners_count":20361378,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":[],"created_at":"2024-11-22T12:26:41.052Z","updated_at":"2025-10-15T04:36:44.348Z","avatar_url":"https://github.com/feifeibear.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Dsirtibuted ResNet in TensorFlow\nThis repo implements distributed resnet with horovod\n\nDeep residual networks, or ResNets for short, provided the breakthrough idea of identity mappings in order to enable training of very deep convolutional neural networks. This folder contains an implementation of ResNet for the ImageNet dataset written in TensorFlow.\n\nSee the following papers for more background:\n\n[Deep Residual Learning for Image Recognition](https://arxiv.org/pdf/1512.03385.pdf) by Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun, Dec 2015.\n\n[Identity Mappings in Deep Residual Networks](https://arxiv.org/pdf/1603.05027.pdf) by Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun, Jul 2016.\n\nPlease proceed according to which dataset you would like to train/evaluate on:\n\n\n## CIFAR-10\n\n### Setup\n\nYou simply need to have the latest version of TensorFlow installed.\n\nFirst download and extract the CIFAR-10 data from Alex's website, specifying the location with the `--data_dir` flag. Run the following:\n\n```\npython cifar10_download_and_extract.py\n```\n\nThen to train the model, run the following:\n\n```\npython cifar10_main.py\n```\n\nUse `--data_dir` to specify the location of the CIFAR-10 data used in the previous step. There are more flag options as described in `cifar10_main.py`.\n\n\n## ImageNet\n\n### Setup\nTo begin, you will need to download the ImageNet dataset and convert it to TFRecord format. Follow along with the [Inception guide](https://github.com/tensorflow/models/tree/master/research/inception#getting-started) in order to prepare the dataset.\n\nOnce your dataset is ready, you can begin training the model as follows:\n\n```\npython imagenet_main.py --data_dir=/path/to/imagenet\n```\n\nThe model will begin training and will automatically evaluate itself on the validation data roughly once per epoch.\n\nNote that there are a number of other options you can specify, including `--model_dir` to choose where to store the model and `--resnet_size` to choose the model size (options include ResNet-18 through ResNet-200). See [`imagenet_main.py`](imagenet_main.py) for the full list of options.\n\n### Pre-trained model\nYou can download a 190 MB pre-trained version of ResNet-50 achieving 75.3% top-1 single-crop accuracy here: [resnet50_2017_11_30.tar.gz](http://download.tensorflow.org/models/official/resnet50_2017_11_30.tar.gz). Simply download and uncompress the file, and point the model to the extracted directory using the `--model_dir` flag.\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Ffeifeibear%2Fhorovod-resnet","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Ffeifeibear%2Fhorovod-resnet","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Ffeifeibear%2Fhorovod-resnet/lists"}