modify RAdam

2022-04-01 21:44:53 +08:00 · 2022-04-01 21:44:53 +08:00 · 843cd8b98f
commit 843cd8b98f
parent dacf9bc583
1 changed files with 1 additions and 1 deletions
--- a/keras_textclassification/m01_FastText/graph.py
+++ b/keras_textclassification/m01_FastText/graph.py
@ -144,7 +144,7 @@ class FastTextGraph(graph):
          构建优化器、损失函数和评价函数
        :return:
        """
-        from keras_textclassification.keras_layers.keras_radam import RAdam
+        # from keras_textclassification.keras_layers.keras_radam import RAdam
        from keras.optimizers import Adam
        # self.model.compile(optimizer=Adam(lr=self.lr, beta_1=0.9, beta_2=0.999, decay=0.0),
        #                    loss=[self.focal_loss(alpha=.25, gamma=2)],