#

pytorch

  • 解決Pytorch 訓(xùn)練與測試時爆顯存(out of memory)的問題

    Pytorch 訓(xùn)練時有時候會因?yàn)榧虞d的東西過多而爆顯存，有些時候這種情況還可以使用cuda的清理技術(shù)進(jìn)行修整，當(dāng)然如果模型實(shí)在太大，那也沒辦法。 使用torch.cuda.empty_cache()

    作者：xiaoxifei
    2020-09-23 06:29:08

  • pytorch 更改預(yù)訓(xùn)練模型網(wǎng)絡(luò)結(jié)構(gòu)的方法

    一個繼承nn.module的model它包含一個叫做children()的函數(shù)，這個函數(shù)可以用來提取出model每一層的網(wǎng)絡(luò)結(jié)構(gòu)，在此基礎(chǔ)上進(jìn)行修改即可，修改方法如下(去除后兩層)： resnet

    作者：wayne980
    2020-09-22 17:43:02

  • Windows下PyTorch開發(fā)環(huán)境安裝教程

    Anaconda安裝   Anaconda是為方便使用python而建立的一個軟件包，其包含常用的250多個工具包，多版本python解釋器和強(qiáng)大的虛擬環(huán)境管理工具，所以Ana

    作者：Sakura櫻_子于
    2020-09-22 14:08:14

  • pytorch中tensor張量數(shù)據(jù)類型的轉(zhuǎn)化方式

    1.tensor張量與numpy相互轉(zhuǎn)換 tensor ----->numpy import torch a=torch.ones([2,5]) tensor([[1., 1., 1.,

    作者：|晴天|
    2020-09-21 14:31:17

  • Pytorch 保存模型生成圖片方式

    三通道數(shù)組轉(zhuǎn)成彩色圖片 img=np.array(img1) img=img.reshape(3,img1.shape[2],img1.shape[3]) img=(i

    作者：wzy_zju
    2020-09-21 11:58:07

  • pytorch制作自己的LMDB數(shù)據(jù)操作示例

    本文實(shí)例講述了pytorch制作自己的LMDB數(shù)據(jù)操作。分享給大家供大家參考，具體如下： 前言 記錄下pytorch里如何使用lmdb的code，自用 制作部分的Code code就是ASTER里數(shù)據(jù)

    作者：團(tuán)長sama
    2020-09-20 18:06:31

  • pytorch 可視化feature map的示例代碼

    之前做的一些項(xiàng)目中涉及到feature map 可視化的問題，一個層中feature map的數(shù)量往往就是當(dāng)前層out_channels的值，我們可以通過以下代碼可視化自己網(wǎng)絡(luò)中某層的feature

    作者：牛丸4
    2020-09-20 00:15:36

  • pytorch GAN偽造手寫體mnist數(shù)據(jù)集方式

    一，mnist數(shù)據(jù)集 形如上圖的數(shù)字手寫體就是mnist數(shù)據(jù)集。 二，GAN原理(生成對抗網(wǎng)絡(luò)) GAN網(wǎng)絡(luò)一共由兩部分組成：一個是偽造器(Generator，簡稱G)，一個是判別器(Discrim

    作者：ZJE_ANDY
    2020-09-19 11:41:00

  • 淺析PyTorch中nn.Linear的使用

    查看源碼 Linear 的初始化部分： class Linear(Module): ... __constants__ = ['bias'] def __init__(self, in

    作者：Steven·簡談
    2020-09-18 20:40:25

  • 基于pytorch的lstm參數(shù)使用詳解

    lstm(*input, **kwargs) 將多層長短時記憶(LSTM)神經(jīng)網(wǎng)絡(luò)應(yīng)用于輸入序列。 參數(shù)： input_size:輸入'x'中預(yù)期特性的數(shù)量 hidden_size:隱藏狀態(tài)'h'中的

    作者：hufei_neo
    2020-09-17 22:46:15
< 上一頁
下一頁>