Release FuxiCTR v2.3.4

reczoo · Nov 5, 2024 · dba7239 · dba7239
2 parents 0996f7e + 25c03a2
commit dba7239
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 2 deletions.
diff --git a/model_zoo/WuKong/config/model_config.yaml b/model_zoo/WuKong/config/model_config.yaml
@@ -1,6 +1,10 @@
 Base:
     model_root: './checkpoints/'
+<<<<<<< HEAD
     num_workers: 3
+=======
+    num_workers: 12
+>>>>>>> 25c03a2e178971a94fc718bcd5f086ad8c0737e8
     verbose: 1
     early_stop_patience: 2
     pickle_feature_encoder: True
@@ -58,8 +62,16 @@ WuKong_test:
     dropout_rate: 0.2
     hidden_activations: relu
     mlp_hidden_units: [64]
+<<<<<<< HEAD
     epochs: 1
     shuffle: True
     seed: 2024
     monitor: 'AUC'
     monitor_mode: 'max'
+=======
+    epochs: 5
+    shuffle: True
+    seed: 2024
+    monitor: 'AUC'
+    monitor_mode: 'max'
+>>>>>>> 25c03a2e178971a94fc718bcd5f086ad8c0737e8
diff --git a/model_zoo/WuKong/src/WuKong.py b/model_zoo/WuKong/src/WuKong.py
@@ -133,7 +133,7 @@ def forward(self, x):
 
 
 class LinearCompressionBlock(nn.Module):
-    # Linear Compression Block (LCB)
+    """ Linear Compression Block (LCB) """
     def __init__(self, num_features=14, embedding_dim=16, compressed_dim=8,dropout_rate=0.2):
         super(LinearCompressionBlock, self).__init__()
         self.linear = nn.Linear(num_features * embedding_dim, compressed_dim)
@@ -143,7 +143,6 @@ def forward(self, x):
 
 
 class WuKongLayer(nn.Module):
-    # WuKong Layer
     def __init__(self, num_features=14, embedding_dim=16, project_dim=4, fmb_units=[40,40,40], fmb_dim=40, compressed_dim=40, dropout_rate=0.2):
         super(WuKongLayer, self).__init__()
         self.fmb = FMB(num_features, embedding_dim, fmb_units, fmb_dim, project_dim)

diff --git a/model_zoo/WuKong/src/__init__.py b/model_zoo/WuKong/src/__init__.py
@@ -1 +1,5 @@
+<<<<<<< HEAD
 from .WuKong import *
+=======
+from .WuKong import *
+>>>>>>> 25c03a2e178971a94fc718bcd5f086ad8c0737e8