razmars commited on
Commit
6a3442d
·
verified ·
1 Parent(s): 7179059

Update configuration_super_linear.py

Browse files
Files changed (1) hide show
  1. configuration_super_linear.py +13 -13
configuration_super_linear.py CHANGED
@@ -50,17 +50,17 @@ class SuperLinearConfig(PretrainedConfig):
50
  self.layer_type = "RLinear"
51
  self.linear_checkpoints_path = '/cs/azencot_fsas/MoE/'
52
  self.linear_checkpoints_dir = "checkpoints5"
53
- self.load_linear = 0
54
- self.manual_moe = 0
55
- self.misc_moe = 1
56
- self.noisy_gating_std = 0.1
57
- self.noisy_gating_std_decay = 1
58
- self.ker_len = 50
59
- self.con = 0
60
- self.d_model = 128
61
- self.mlp_gating = 1
62
- self.moe_temp = 1
63
- self.use_fft = 1
64
- self.fft_len = 10000
65
- self.dropout = 0.0
66
  super().__init__(**kwargs)
 
50
  self.layer_type = "RLinear"
51
  self.linear_checkpoints_path = '/cs/azencot_fsas/MoE/'
52
  self.linear_checkpoints_dir = "checkpoints5"
53
+ self.load_linear = load_linear
54
+ self.manual_moe = manual_moe
55
+ self.misc_moe = misc_moe
56
+ self.noisy_gating_std = noisy_gating_std
57
+ self.noisy_gating_std_decay = noisy_gating_std_decay
58
+ self.ker_len = ker_len
59
+ self.con = con
60
+ self.d_model = d_model
61
+ self.mlp_gating = mlp_gating
62
+ self.moe_temp = moe_temp
63
+ self.use_fft = use_fft
64
+ self.fft_len = fft_len
65
+ self.dropout = dropout
66
  super().__init__(**kwargs)