'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()
'device': '/gpu:1',
#####ISTA PARAMS######
'numIterations': 100000,
'displayPeriod': 200,
#Batch size
'batchSize': 32,
#Learning rate for optimizer
'learningRateA': 1e-3,
'learningRateW': 1e-4,
#Lambda in energy function
'thresh': .015,
#Number of features in V1
'numV': 128,
#Stride of V1
'VStrideY': 2,
'VStrideX': 2,
#Patch size
'patchSizeY': 12,
'patchSizeX': 12,
}
#Allocate tensorflow object
tfObj = ISTA(params, trainDataObj)
print "Done init"
tfObj.writePvpWeights(outPvpPrefix, rect=False)
print "Done run"
tfObj.closeSess()
#Device to run on
'device': '/gpu:1',
#####ISTA PARAMS######
'numIterations': 100000,
'displayPeriod': 200,
#Batch size
'batchSize': 32,
#Learning rate for optimizer
'learningRateA': 1e-3,
'learningRateW': 1e-4,
#Lambda in energy function
'thresh': .015,
#Number of features in V1
'numV': 128,
#Stride of V1
'VStrideY': 2,
'VStrideX': 2,
#Patch size
'patchSizeY': 12,
'patchSizeX': 12,
}
#Allocate tensorflow object
tfObj = ISTA(params, trainDataObj)
print "Done init"
tfObj.writePvpWeights(outPvpPrefix, rect=False)
print "Done run"
tfObj.closeSess()