razmars commited on
Commit
ef55773
·
verified ·
1 Parent(s): fc6971b

Update modeling_super_linear.py

Browse files
Files changed (1) hide show
  1. modeling_super_linear.py +5 -5
modeling_super_linear.py CHANGED
@@ -416,11 +416,11 @@ class superLinear(nn.Module):
416
  # for param in self.experts[expert_freq].parameters():
417
  # param.requires_grad = False
418
 
419
- self.n_experts = len(self.experts)
420
- else:
421
- for i in range(self.n_experts):
422
- print(f"creating expert {i}")
423
- self.experts[str(i)] = self.layer_dict[self.layer_type](self.seq_len, self.pred_len)
424
 
425
 
426
  if configs.misc_moe>0:
 
416
  # for param in self.experts[expert_freq].parameters():
417
  # param.requires_grad = False
418
 
419
+ # self.n_experts = len(self.experts)
420
+ # else:
421
+ # for i in range(self.n_experts):
422
+ # print(f"creating expert {i}")
423
+ # self.experts[str(i)] = self.layer_dict[self.layer_type](self.seq_len, self.pred_len)
424
 
425
 
426
  if configs.misc_moe>0: