From d7c5d57d934a1e78c9607d2dc237bc795210d2a4 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Wed, 22 Sep 2021 12:10:17 +0200 Subject: [PATCH] Decreased blob-size a bit to make training faster --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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