Skip to content

Feature/docker#42

Open
motokimura wants to merge 11 commits intobrazgardian:masterfrom
motokimura:feature/docker
Open

Feature/docker#42
motokimura wants to merge 11 commits intobrazgardian:masterfrom
motokimura:feature/docker

Conversation

@motokimura
Copy link
Copy Markdown

Hi @garrickbrazil, thank you for making the code public.

I added Dockerfile to this repo.
I hope this helps someone who cannot prepare old CUDA environment in local.

For usage, please see docker/README.md.

@motokimura
Copy link
Copy Markdown
Author

motokimura commented Aug 2, 2020

Note:

  • I've checked the pre-trained models provided by garrickbrazil could reproduce mAP with this docker environment.
  • I've not tested the models trained with the docker could reproduce the results.

Reports from someone tried training are more than welcome.
I used the same versions for pytorch, torchvision, and other libraries (see docker/requirements.txt) so I hope it reproduces training as well.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant