feat: model --> ready to test train

This commit is contained in:
Robin Meersman 2025-11-08 20:55:05 +01:00
parent 63d1b6f5ae
commit b58682cb49
8 changed files with 382 additions and 17 deletions

View file

@ -14,7 +14,6 @@ class CNNPredictor(nn.Module):
def __init__(
self,
vocab_size=256,
context_length=128,
num_layers=3,
hidden_dim=128,
kernel_size=3,