search

perrigoh / image_classifier_pytorch_udacity

Image classifier model using PyTorch
MIT License
0 stars 0 forks source link

use GPU if is available #4

Closed perrigoh closed 1 year ago

perrigoh commented 1 year ago
  1. gpu argument need to remove
  2. change gpu loop to if is_available
  3. remove in_arg.gpu argument from predict(in_arg.save_dir, in_arg.gpu, in_arg.topk, in_arg.category_names) note: previously same fix done to other files remember to check the SHA and combine the commits

    commit a5db98b6c3b9781a72308417e086040969d76772

perrigoh commented 1 year ago

1 to 3 done.

note: didn't combine this commit to commit a5db98b6c3b9781a72308417e086040969d76772 as changes made without creating a new branch, hence not able to use git merge.