base_dir = "/content/dog vs cat/dataset/training_set" batch_size = 64 __ __