运行: cd /home/peace/Downloads/ conda activate snn
直接运行: python ./BSNN/main.py
后台运行: nohup python ./BSNN/main.py > ./BSNN/main.log 2>&1 &
选择显卡,使用: os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'
限制总可用资源,使用: ray.init(num_cpus=16, num_gpus=2,)
关于参数: config = { 'binarize_weight':tune.grid_search(['ste','xnor','none']), 'norm_method':tune.grid_search(['BN','BNTT','tdBN','TEBN']), 'binarize_norm':tune.grid_search([True,False]), 'pool_method':tune.grid_search(['max','avg']), 'net_structure':tune.grid_search(['CBNPBN','CBNPXN','CBNPXX','CXXPBN']) } 'binarize_weight'---二值化权重: 'ste'---使用ste进行二值化 'xnor'----使用xnor进行二值化 'none'---不进行二值化
'norm_method'---norm方法选择: 'BN'---普通BatchNorm 'BNTT'--- 'tdBN'--- 'TEBN'---
'binarize_norm'---是否使用shiftBN: 'True'---开启 'False'---关闭
'pool_method'---pool的方法: 'max'--maxpooling 'avg'--avgpooling
net_structure--网络结构: 'C'---卷积 'B'---Norm 'N'---neuron 'P'---pooling 'X'---空(None)