Spaces:
Runtime error
Runtime error
File size: 368 Bytes
e4497d1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import os
DATASET_URL = 'https://download.microsoft.com/download/3/E/1/3E1C3F21-ECDB-4869-8368-6DEBA77B919F/kagglecatsanddogs_3367a.zip'
CACHE_DIR = os.getcwd()
CACHE_SUBDIR = 'data'
if not os.path.isdir(CACHE_SUBDIR):
os.mkdir(CACHE_SUBDIR)
DATASET_PATH = os.path.join(CACHE_DIR, CACHE_SUBDIR, 'PetImages')
IMAGE_SIZE = (180, 180)
BATCH_SIZE = 32
EPOCHS = 50 |