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

Display changes and CPU compatibility #3

Open

prateekmalhotra wants to merge 4 commits into valencebond:master from prateekmalhotra:master

prateekmalhotra commented May 26, 2020

The training process per epoch is quite slow even on a Tesla V100, so I thought it's best to display a progress bar using tqdm
Added .to(device) everywhere instead of .cuda() so people with CPU instances can have a go at the code if they like.

Prateek Malhotra and others added 3 commits

May 25, 2020 17:17


          Added CPU functionality

c78adc5


          Update train.py

f612990


          Better display

01ae03d

prateekmalhotra changed the title ~~Display changes and made the code CPU compatible~~ Display changes and CPU compatibility


          Added DeepWeeds dataset

cb4d4b3

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

Labels

None yet