Decreased blob-size a bit to make training faster
This commit is contained in:
parent
ca964732c5
commit
d7c5d57d93
2
train.py
2
train.py
@ -9,7 +9,7 @@ import math
|
|||||||
import shark
|
import shark
|
||||||
from model import Model
|
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)
|
tid = str(int(random.random()*99999)).zfill(5)
|
||||||
print("[i] I am "+str(tid))
|
print("[i] I am "+str(tid))
|
||||||
ltLoss = 0.75
|
ltLoss = 0.75
|
||||||
|
Loading…
Reference in New Issue
Block a user