Update chess-gpt-eval-contrastive/mamba_module.py
Browse files
chess-gpt-eval-contrastive/mamba_module.py
CHANGED
|
@@ -8,6 +8,8 @@ import numpy as np
|
|
| 8 |
from functools import partial
|
| 9 |
import chess
|
| 10 |
from sklearn.linear_model import LinearRegression
|
|
|
|
|
|
|
| 11 |
|
| 12 |
BASE_DIR = "mamba/"
|
| 13 |
|
|
|
|
| 8 |
from functools import partial
|
| 9 |
import chess
|
| 10 |
from sklearn.linear_model import LinearRegression
|
| 11 |
+
import torch.nn as nn
|
| 12 |
+
import torch.optim as optim
|
| 13 |
|
| 14 |
BASE_DIR = "mamba/"
|
| 15 |
|