Python FISTA Examples

Example #1

    #####ISTA PARAMS######
    'numIterations': 6250,
    '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, testDataObj)
print "Done init"
outFilename = params["outDir"] + params[
    "runDir"] + "fista_test_cifar_256_eval.pvp"

tfObj.evalSet(testDataObj, outFilename)
print "Done run"

tfObj.closeSess()

Example #2

File: fista_cifar_train.py Project: slundqui/TFSparseCode

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

Example #3

File: fista_cifar_traineval.py Project: slundqui/TFSparseCode

    #####ISTA PARAMS######
    'numIterations':   6250,
    '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"
outFilename = params["outDir"] + params["runDir"] + "fista_train_cifar_256_eval.pvp"

tfObj.evalSet(trainDataObj, outFilename)
print "Done run"

tfObj.closeSess()

Example #4

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