New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

bump albumentations; support resuming of training #42

Open

dvschultz wants to merge 2 commits into NVlabs:master from dvschultz:master

dvschultz commented Dec 7, 2020

This PR does two things:

Requires a minimum version of albumentations (fixes an issue I found with running on Colab)
Allows you to resume training using the --resume argument. The code to resume training already existed in this repo but by default was always set to None. Passing in --resume with the train.py script now sets it to True

dvschultz added 2 commits

November 23, 2020 16:12


          bump albumentations for colab

832b1cc


          support resuming training

1ac4784

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

Labels

None yet