'device': '/gpu:0', #####FISTA PARAMS###### 'numIterations': 100000, 'displayPeriod': 200, #Batch size 'batchSize': 8, #Learning rate for optimizer 'learningRateA': .01, 'learningRateW': 1, #Lambda in energy function 'thresh': .005, #Number of features in V1 'numV': 256, #Stride of V1 'VStrideY': 2, 'VStrideX': 2, #Patch size 'patchSizeY': 12, 'patchSizeX': 12, } #Allocate tensorflow object tfObj = FISTA(params, trainDataObj) print("Done init") tfObj.runModel() print("Done run") tfObj.closeSess()
#Device to run on 'device': '/gpu:0', #####FISTA PARAMS###### 'numIterations': 100000, 'displayPeriod': 200, #Batch size 'batchSize': 8, #Learning rate for optimizer 'learningRateA': .01, 'learningRateW': 1, #Lambda in energy function 'thresh': .005, #Number of features in V1 'numV': 256, #Stride of V1 'VStrideY': 2, 'VStrideX': 2, #Patch size 'patchSizeY': 12, 'patchSizeX': 12, } #Allocate tensorflow object tfObj = FISTA(params, trainDataObj) print("Done init") tfObj.runModel() print("Done run") tfObj.closeSess()