Tuning: Added AlexNet sizes for GEMM

This commit is contained in:
Philippe Tillet
2015-07-16 14:26:21 -04:00
parent 1e3c853b58
commit 3d4c9cf1e2
3 changed files with 28 additions and 7 deletions

View File

@@ -31,9 +31,10 @@ def train(X, Y, profiles):
Y = Y[p,:]
#Train the model
cut = int(0.9*M)
XTr, YTr = X[:cut,:], Y[:cut,:]
XCv, YCv = X[cut:,:], Y[cut:,:]
cut = int(1.00*M)
CV = .1
XTr, YTr = X[:,:], Y[:,:]
XCv, YCv = X[:max(1,CV*M),:], Y[:max(1,CV*M),:]
nrmses = {}
for N in range(1,min(M+1,20)):