caffe
关注
分享
扫一扫
Yao_tc
这个作者很懒，什么都没留下…
展开
-
Singularity配置Pytorch0.4.1 + Anaconda3（Python3.5） + CUDA9.0 + CUDNN7.0
Singularity recipeBootstrap: dockerFrom:nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04%post echo 'nameserver 8.8.8.8' >> /etc/resolv.conf apt-get update apt-get install -y wget ...原创 2018-10-30 15:05:45 · 1361 阅读 · 3 评论 -
将caffe预训练模型的权重载入pytorch
#!/usr/bin/env python2.7#coding=utf-8import caffeimport csvimport numpy as np# np.set_printoptions(threshold='nan')MODEL_FILE = 'inception_v3_rgb_deploy.prototxt'PRETRAIN_FILE = 'inception_v3...原创 2018-12-04 15:51:11 · 3064 阅读 · 0 评论