quan0401 commited on
Commit
77048ae
·
1 Parent(s): fb175e0

change defeault pytorch device to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ __all__ = ['path', 'files', 'dls', 'learn', 'categories', 'label_func', 'image_c
7
  from fastai.vision.all import ImageDataLoaders, URLs, untar_data, get_image_files, Resize, \
8
  vision_learner, resnet34, error_rate, default_device, load_learner, show_image, PILImage
9
  import torch
 
10
 
11
  # %% cat_classifier.ipynb 3
12
  path = untar_data(URLs.PETS)
 
7
  from fastai.vision.all import ImageDataLoaders, URLs, untar_data, get_image_files, Resize, \
8
  vision_learner, resnet34, error_rate, default_device, load_learner, show_image, PILImage
9
  import torch
10
+ torch.set_default_device('cpu')
11
 
12
  # %% cat_classifier.ipynb 3
13
  path = untar_data(URLs.PETS)