原创 pytorch 記錄

這裏寫自定義目錄標題pytorch 加載部分權重pytorch 固定權重 pytorch 加載部分權重 pretrained_dict = torch.load('{0}/{1}'.format(opt.outf, opt.res

原创 docker,nvidia-docker,labelfusion安裝

docker 按官方的教程 sudo apt-get update sudo apt-get install docker-ee docker-ee-cli containerd.io sudo docker run hello-

原创 pytorch代碼

模型加載,從cpu到gpu self.net.load_state_dict(torch.load(net_path,map_location=lambda storage,loc:storage)) namedetuple給元

原创 數組模擬鄰接表

#include<iostream> #include<algorithm> #include<string.h> using namespace std; const int N = 10010,M = 20010; int

原创 R-CNN系列

R-cnn 基礎的網絡架構設計,先提取框,再在框上進行卷積。 fast R-cnn 直接卷積,在feature map 上卷積 faster R-cnn 使用RPN方法進行選框,end-to-end mask R-cnn 加條路徑

原创 cuda,anaconda安裝

cuda安裝 安裝依賴 sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev prot

原创 conda環境遷移(離線)

離線環境遷移 conda 版本一致,直接複製anaconda/envs/env_name整個文件夾,到新環境的對應位置,然後: conda activate env_name 即可 若conda env list 裏面沒有對應環

原创 線段樹

#include <string.h> #include <vector> #include <algorithm> #include <bits/stdc++.h> using namespace std; const int

原创 pytorch疑問或bug

0.2 pytorch0.2版本中申請顯存會非常慢。