Update modeling_super_linear.py
Browse files- modeling_super_linear.py +0 -1
modeling_super_linear.py
CHANGED
|
@@ -387,7 +387,6 @@ class SparseNoisyMoE(nn.Module):
|
|
| 387 |
if hasattr(self, 'experts') and isinstance(getattr(self, 'experts', None), dict):
|
| 388 |
# If experts are stored in a dictionary with meaningful keys
|
| 389 |
expert_names = list(self.experts.keys())
|
| 390 |
-
print(expert_names)
|
| 391 |
plt.xticks(range(len(probs_np)), expert_names, rotation=45, ha='right')
|
| 392 |
else:
|
| 393 |
# Default numbering if expert names aren't available
|
|
|
|
| 387 |
if hasattr(self, 'experts') and isinstance(getattr(self, 'experts', None), dict):
|
| 388 |
# If experts are stored in a dictionary with meaningful keys
|
| 389 |
expert_names = list(self.experts.keys())
|
|
|
|
| 390 |
plt.xticks(range(len(probs_np)), expert_names, rotation=45, ha='right')
|
| 391 |
else:
|
| 392 |
# Default numbering if expert names aren't available
|