Train on bigger blobs (1KiB)

2021-09-22 11:46:22 +02:00 · 2021-09-22 11:46:22 +02:00 · 8608a89909
commit 8608a89909
parent e4e2a18606
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -9,7 +9,7 @@ import math
 import shark
 from model import Model

-def train(model, seq_len=16*64):
+def train(model, seq_len=16*512): # 1KiB
    tid = str(int(random.random()*99999)).zfill(5)
    print("[i] I am "+str(tid))
    ltLoss = 0.75