Amui 次元

我知这世界,如露水般短暂,然而,然而

Python

map 函数 map()是python内置的高阶函数,对传入的list的每一个元素进行映射,返回一个新的映射。 对于python2.x,映射后的值为list; 对于python3.x,映射后的值是一个map对象如:<map object at 0x7f1234>。若要重新得到list,需要外层在嵌一层list()。 python2.x示例: def f(x): re...

Densenet

DenseNet( (features): Sequential( (init_conv): Conv2d(3, 24, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False) (denseblock_1): _DenseBlock( (denselayer_1): _Den...

Pytorch

Pytorch-神经网络相关记录 torch.manual_seed 在训练开始时,参数的初始化是随机的,为了让每次的结果一致,我们需要设置随机种子。 torch.manual_seed(args.seed)#为CPU设置随机种子 torch.cuda.manual_seed(seed) #为当前GPU设置随机种子 torch.cuda.manual_seed_all(seed) #...

tensorflow指定使用的GPU

如果服务器有多个GPU,tensorflow会默认使用所有能用的GPU,若想只使用部分GPU,可以设置CUDA_VISIBLE_DEVICES。 1. 命令行设置使用的GPU 在命令行执行python脚本时,可以如下指定要使用的GPU: CUDA_VISIBLE_DEVICES=1 python my_script.py 指定不同GPU方式如下: CUDA_VISIBLE_...

神经网络 —— 模型压缩

模型压缩

摘要 模型压缩