#Num iterations
    'outerSteps': 10000000,  #1000000,
    'innerSteps': 50,  #300,
    #Batch size
    'batchSize': 128,
    #Learning rate for optimizer
    'learningRate': 1e-3,
    'numClasses': 10,
    'verifyTrain': False,
    'verifyTest': False,

    #####ISTA PARAMS######
    'VStrideY': 2,
    'VStrideX': 2,
    'rectify': False,

    #SLP params
    'pooledY': 8,
    'pooledX': 8,
}

#Allocate tensorflow object
#This will build the graph
tfObj = SLP(params, trainDataObj.inputShape)

print "Done init"
tfObj.runModel(trainDataObj, testDataObj=testDataObj)
print "Done run"

tfObj.closeSess()