Train on bigger blobs (1KiB)
This commit is contained in:
parent
e4e2a18606
commit
8608a89909
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*64):
|
def train(model, seq_len=16*512): # 1KiB
|
||||||
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