missing import

This commit is contained in:
Starbeamrainbowlabs 2023-01-05 18:26:33 +00:00
parent 0d4cc63b76
commit 6a4f68a055
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -15,6 +15,8 @@ import matplotlib.pyplot as plt
import tensorflow as tf import tensorflow as tf
from lib.dataset.dataset_mono import dataset_mono
IMAGE_SIZE = int(os.environ["IMAGE_SIZE"]) if "IMAGE_SIZE" in os.environ else 128 # was 512; 128 is the highest power of 2 that fits the data IMAGE_SIZE = int(os.environ["IMAGE_SIZE"]) if "IMAGE_SIZE" in os.environ else 128 # was 512; 128 is the highest power of 2 that fits the data
BATCH_SIZE = int(os.environ["BATCH_SIZE"]) if "BATCH_SIZE" in os.environ else 64 BATCH_SIZE = int(os.environ["BATCH_SIZE"]) if "BATCH_SIZE" in os.environ else 64
NUM_CLASSES = 2 NUM_CLASSES = 2