Decreased blob-size a bit to make training faster

2021-09-22 12:10:17 +02:00 · 2021-09-22 12:10:17 +02:00 · d7c5d57d93
commit d7c5d57d93
parent ca964732c5
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*512): # 1KiB
+def train(model, seq_len=16*128): # 0.25KiB
    tid = str(int(random.random()*99999)).zfill(5)
    print("[i] I am "+str(tid))
    ltLoss = 0.75