瀏覽代碼

Add standard experts

Max Ryabinin 2 年之前
父節點
當前提交
0fc36c8eed
共有 1 個文件被更改,包括 0 次插入1 次删除
  1. 0 1
      hivemind/moe/server/layers/albert.py

+ 0 - 1
hivemind/moe/server/layers/albert.py

@@ -571,7 +571,6 @@ class HeadExpert(nn.Module):
 class StandardHeadExpert(HeadExpert):
     def __init__(self, hid_dim):
         super().__init__(hid_dim)
-        self.config.num_hidden_layers = 1
 
 
 body_sample_input = lambda batch_size, hid_dim: (torch.empty((batch_size, SEQUENCE_LENGTH, hid_dim)),)