Python ISTA示例

示例#1

文件： imagenet_train.py 项目： slundqui/TFSparseCode

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

示例#2

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

示例#3

文件： cifar_writepvp.py 项目： slundqui/TFSparseCode

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

示例#4

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