#output plots directory 'plotDir': "plots/", 'plotPeriod': 20, #With respect to displayPeriod #Progress step (also controls how often to save and write out to tensorboard) 'progress': 300, #####ISTA PARAMS###### 'batchSize': 1, #Learning rate for optimizer 'learningRateA': 1e-4, 'learningRateW': 1e-4, #Lambda in energy function 'thresh': .0125, #Number of features in V1 'numV': 3072, #Stride of V1 'VStride': 4, #Patch size 'patchSizeY': 16, 'patchSizeX': 32, } #Allocate tensorflow object #This will build the graph tfObj = ISTA(params, trainDataObj) print("Done init") tfObj.runModel() print("Done run") tfObj.closeSess()