diff --git a/train.py b/train.py index 69e819e..a915aaf 100644 --- 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