Spaces:
Runtime error
Runtime error
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 |