- 博客(3)
- 资源 (7)
- 收藏
- 关注
原创 pytorh resnet网络代码分析
1.基本块 class Bottleneck(nn.Module): expansion = 4 def __init__(self, inplanes, planes, stride=1, downsample=None): super(Bottleneck, self).__init__() self.conv1 = nn.Conv2d(i...
2019-08-21 23:41:44 318
原创 pytorch安装对应版本torchvision
1.代码要求版本1.0,pytorch1.1会报错 ImportError: /usr/local/lib/python3.5/dist-packages/torchvision/_C.cpython-35m-x86_64-linux-gnu.so: undefined symbol: _ZN2at7getTypeERKNS_6TensorE 2. 安装torch1.0 要对应安装torch...
2019-08-17 15:37:21 53802 3
原创 keras 版本PSEnet训练过程记录
1.由分步执行改成一个文件 训练文件 import keras.backend.tensorflow_backend as KTF import tensorflow as tf config = tf.ConfigProto() config.gpu_options.allow_growth=True session = tf.Session(config=config) KTF.set_...
2019-08-09 20:09:47 1267 2
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人