Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ tf.random.set_seed(42)
|
|
7 |
|
8 |
# Initialize global variables.
|
9 |
AUTO = tf.data.AUTOTUNE
|
10 |
-
BATCH_SIZE =
|
11 |
NUM_SAMPLES = 32
|
12 |
POS_ENCODE_DIMS = 16
|
13 |
EPOCHS = 30
|
|
|
7 |
|
8 |
# Initialize global variables.
|
9 |
AUTO = tf.data.AUTOTUNE
|
10 |
+
BATCH_SIZE = 1
|
11 |
NUM_SAMPLES = 32
|
12 |
POS_ENCODE_DIMS = 16
|
13 |
EPOCHS = 30
|