'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()
#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()
