crobbi commited on
Commit
fbf52e6
1 Parent(s): edb390c

Update modelutil.py

Browse files
Files changed (1) hide show
  1. modelutil.py +6 -6
modelutil.py CHANGED
@@ -12,12 +12,12 @@ def create_model():
12
  model.load_weights('./checkpoint')
13
 
14
 
15
- LOSS_FUNCTION = tf.keras.losses.SparseCategoricalCrossentropy() # HERE
16
- OPTIMIZER = tf.keras.optimizers.legacy.Adam()
17
- METRICS = ["accuracy"]
18
- model.compile(loss=LOSS_FUNCTION,
19
- optimizer=OPTIMIZER,
20
- metrics=METRICS)
21
 
22
  return model
23
 
 
12
  model.load_weights('./checkpoint')
13
 
14
 
15
+ # LOSS_FUNCTION = tf.keras.losses.SparseCategoricalCrossentropy() # HERE
16
+ # OPTIMIZER = tf.keras.optimizers.legacy.Adam()
17
+ # METRICS = ["accuracy"]
18
+ # model.compile(loss=LOSS_FUNCTION,
19
+ # optimizer=OPTIMIZER,
20
+ # metrics=METRICS)
21
 
22
  return model
23