modify Radam

2022-04-01 21:47:59 +08:00 · 2022-04-01 21:47:59 +08:00 · edcce1b2ca
commit edcce1b2ca
parent 2997b692f5
1 changed files with 1 additions and 1 deletions
--- a/test/tft_tensorflow_transformer/graph.py
+++ b/test/tft_tensorflow_transformer/graph.py
@ -57,7 +57,7 @@ class TextCNNGraphTFT(graph):
        self.path_fineture = hyper_parameters_model.get('path_fineture',
                                                        path_fineture)  # embedding层保存地址, 例如静态词向量、动态词向量、微调bert层等
        self.patience = hyper_parameters_model.get('patience', 3)  # 早停, 2-3就可以了
-        self.optimizer_name = hyper_parameters_model.get('optimizer_name', 'RAdam,Lookahead')  # 早停, 2-3就可以了
+        self.optimizer_name = hyper_parameters_model.get('optimizer_name', 'Adam')  # 早停, 2-3就可以了
        if self.gpu_memory_fraction:
            # keras, tensorflow控制GPU使用率等
            import tensorflow as tf