Skip to content

Commit

Permalink
chore: fix typos in main.py
Browse files Browse the repository at this point in the history
  • Loading branch information
nicovandenhooff committed Apr 25, 2022
1 parent bfc18f3 commit ff4b891
Showing 1 changed file with 4 additions and 8 deletions.
12 changes: 4 additions & 8 deletions api/ml/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,13 +4,7 @@
Notes:
- We ran this script in a Kaggle notebook in order to utilize the free GPU.
- The paths to the data set below are absolute paths on Kaggle.
- The beta-version of of Indoor Scene Detector is trained on a subset of this data set
(10/67 classes). We are currently developing support for all 67 classes.
- The "training" module on Kaggle is saved in a Kaggle utility scripts called
"indoorscenes_training", so the import statement looks slightly different
within the Kaggle notebook..
TODO: Add Kaggle notebook link
- TODO: Add Kaggle notebook link
"""

import torch
Expand Down Expand Up @@ -45,8 +39,10 @@
# set device
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

# kaggle absolute path
# kaggle absolute path - full dataset
# DATA_PATH = "../input/indoor-scenes-cvpr-2019/indoorCVPR_09/Images/"

# kaggle absolute path - reduced dataset
DATA_PATH = "../input/top10indoorscenes/reduced-indoor-scenes/images/"

TRANSFORMS = transforms.Compose(
Expand Down

0 comments on commit ff4b891

Please sign in to comment.