replaced all tf function to torch

2023-04-03 08:06:34 +09:00
parent 8bd69bfcd4
commit 57ac1c11d3
3 changed files with 56 additions and 45 deletions
--- a/tools.py
+++ b/tools.py
@@ -548,7 +548,7 @@ class Optimizer():
                            lr=lr,
                            eps=eps),
        'nadam': lambda: NotImplemented(
-                             f'{config.opt} is not implemented'),
+                             f'{opt} is not implemented'),
        'adamax': lambda: torch.optim.Adamax(parameters,
                              lr=lr,
                              eps=eps),