Alesteba commited on
Commit
2e25e60
·
1 Parent(s): e89796e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 = 1
11
  NUM_SAMPLES = 32
12
  POS_ENCODE_DIMS = 16
13
- EPOCHS = 20
14
- H = 50
15
- W = 29
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):