Update chess-gpt-eval-contrastive/mamba_module.py
Browse files
chess-gpt-eval-contrastive/mamba_module.py
CHANGED
|
@@ -11,6 +11,7 @@ from sklearn.linear_model import LinearRegression
|
|
| 11 |
import torch.nn as nn
|
| 12 |
import torch.optim as optim
|
| 13 |
import wandb
|
|
|
|
| 14 |
|
| 15 |
BASE_DIR = "mamba/"
|
| 16 |
|
|
|
|
| 11 |
import torch.nn as nn
|
| 12 |
import torch.optim as optim
|
| 13 |
import wandb
|
| 14 |
+
import math
|
| 15 |
|
| 16 |
BASE_DIR = "mamba/"
|
| 17 |
|