Mr-Vicky-01 commited on
Commit
edffe6d
·
verified ·
1 Parent(s): b41fb04

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -28,8 +28,11 @@ pre_trained_model = Model(inputs=pre_trained.input, outputs=x)
28
  with open("30k_model_architecture.json", "r") as json_file:
29
  loaded_model_json = json_file.read()
30
 
31
- # Redefine the optimizer
32
- optimizer = Adam(lr=0.001)
 
 
 
33
 
34
  # Load weights
35
  model = model_from_json(loaded_model_json)
 
28
  with open("30k_model_architecture.json", "r") as json_file:
29
  loaded_model_json = json_file.read()
30
 
31
+ # Create the optimizer without specifying the learning rate
32
+ optimizer = Adam()
33
+
34
+ # Set the learning rate separately
35
+ optimizer.learning_rate.assign(0.001)
36
 
37
  # Load weights
38
  model = model_from_json(loaded_model_json)