Python AdamSP Examples

Example #1

File: adam_cifar_train.py Project: xueyangfu/TFSparseCode

    #Device to run on
    'device': '/gpu:1',
    #####ISTA PARAMS######
    'numIterations': 1000000,
    'displayPeriod': 200,
    #Batch size
    'batchSize': 32,
    #Learning rate for optimizer
    'learningRateA': 1e-3,
    'learningRateW': 1,
    #Lambda in energy function
    'thresh': .02,
    #Number of features in V1
    'numV': 256,
    #Stride of V1
    'VStrideY': 2,
    'VStrideX': 2,
    #Patch size
    'patchSizeY': 12,
    'patchSizeX': 12,
}

#Allocate tensorflow object
tfObj = AdamSP(params, trainDataObj)
print("Done init")

tfObj.runModel()
print("Done run")

tfObj.closeSess()

Example #2

File: adam_cifar_train.py Project: slundqui/TFSparseCode

    'device':          '/gpu:1',
    #####ISTA PARAMS######
    'numIterations':   1000000,
    'displayPeriod':   200,
    #Batch size
    'batchSize':       32,
    #Learning rate for optimizer
    'learningRateA':   1e-3,
    'learningRateW':   1,
    #Lambda in energy function
    'thresh':          .02,
    #Number of features in V1
    'numV':            256,
    #Stride of V1
    'VStrideY':        2,
    'VStrideX':        2,
    #Patch size
    'patchSizeY':      12,
    'patchSizeX':      12,
}

#Allocate tensorflow object
tfObj = AdamSP(params, trainDataObj)
print "Done init"

tfObj.runModel()
print "Done run"

tfObj.closeSess()

Example #3

File: adam_cifar_testeval.py Project: slundqui/TFSparseCode

    'device':          '/gpu:1',
    #####ISTA PARAMS######
    'numIterations':   6250,
    'displayPeriod':   200,
    #Batch size
    'batchSize':       8,
    #Learning rate for optimizer
    'learningRateA':   1e-3,
    'learningRateW':   1,
    #Lambda in energy function
    'thresh':          .02,
    #Number of features in V1
    'numV':            256,
    #Stride of V1
    'VStrideY':        2,
    'VStrideX':        2,
    #Patch size
    'patchSizeY':      12,
    'patchSizeX':      12,
}

#Allocate tensorflow object
tfObj = AdamSP(params, testDataObj)
print "Done init"
outFilename = params["outDir"] + params["runDir"] + "test_adam_cifar_256_eval.pvp"
tfObj.evalSet(testDataObj, outFilename)
print "Done run"

tfObj.closeSess()

Example #4

File: adam_cifar_testeval.py Project: wen036/TFSparseCode

    'device': '/gpu:1',
    #####ISTA PARAMS######
    'numIterations': 6250,
    'displayPeriod': 200,
    #Batch size
    'batchSize': 8,
    #Learning rate for optimizer
    'learningRateA': 1e-3,
    'learningRateW': 1,
    #Lambda in energy function
    'thresh': .02,
    #Number of features in V1
    'numV': 256,
    #Stride of V1
    'VStrideY': 2,
    'VStrideX': 2,
    #Patch size
    'patchSizeY': 12,
    'patchSizeX': 12,
}

#Allocate tensorflow object
tfObj = AdamSP(params, testDataObj)
print "Done init"
outFilename = params["outDir"] + params[
    "runDir"] + "test_adam_cifar_256_eval.pvp"
tfObj.evalSet(testDataObj, outFilename)
print "Done run"

tfObj.closeSess()