Skip to content

Commit

Permalink
actually test whether adam works
Browse files Browse the repository at this point in the history
  • Loading branch information
arjunsavel committed Mar 11, 2024
1 parent 322ba58 commit 2852799
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/cortecs/fit/fit_neural_net.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,8 @@
try:
# running the legacy Adam optimizer for the Mac folks!
from tensorflow.keras.optimizers.legacy import Adam

Adam(0.1) # test if it works
except ImportError:
from tensorflow.keras.optimizers import Adam
from cortecs.fit.metrics import *
Expand Down

0 comments on commit 2852799

Please sign in to comment.