Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -7,12 +7,12 @@ 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 =
|
14 |
-
H =
|
15 |
-
W =
|
16 |
focal = 138.88
|
17 |
|
18 |
def encode_position(x):
|
|
|
7 |
|
8 |
# Initialize global variables.
|
9 |
AUTO = tf.data.AUTOTUNE
|
10 |
+
BATCH_SIZE = 5
|
11 |
NUM_SAMPLES = 32
|
12 |
POS_ENCODE_DIMS = 16
|
13 |
+
EPOCHS = 30
|
14 |
+
H = 25
|
15 |
+
W = 25
|
16 |
focal = 138.88
|
17 |
|
18 |
def encode_position(x):
|