modify RAdam

2022-04-01 21:44:26 +08:00 · 2022-04-01 21:44:26 +08:00 · dacf9bc583
commit dacf9bc583
parent 6fd54d32fd
1 changed files with 1 additions and 1 deletions
--- a/keras_textclassification/m02_TextCNN/graph.py
+++ b/keras_textclassification/m02_TextCNN/graph.py
@ -160,7 +160,7 @@ class TextCNNGraph(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)],