PyTorch 1 0 中文官方教程 遷移學習教程

2021-09-10 13:58:09 字數 935 閱讀 6543

譯者:片刻

作者: sasank chilamkurthy

在本教程中,您將學習如何使用遷移學習來訓練您的網路。您可以在 cs231n 筆記 上閱讀更多關於遷移學習的資訊

引用這些筆記:

在實踐中,很少有人從頭開始訓練整個卷積網路(隨機初始化),因為擁有足夠大小的資料集是相對罕見的。相反,通常在非常大的資料集(例如 imagenet,其包含具有1000個類別的120萬個影象)上預先訓練 convnet,然後使用 convnet 作為感興趣任務的初始化或固定特徵提取器。

如下是兩個主要的遷移學習場景:

# license: bsd

# author: sasank chilamkurthy

from __future__ import print_function, division

import torch

import torch.nn as nn

import torch.optim as optim

from torch.optim import lr_scheduler

import numpy as np

import torchvision

from torchvision import datasets, models, transforms

import matplotlib.pyplot as plt

import time

import os

import copy

plt.ion(

)# interactive mode

閱讀全文/改進本文

PyTorch 1 0 中文官方教程 神經網路

譯者 bat67 最新版會在譯者倉庫首先同步。可以使用torch.nn包來構建神經網路.我們以及介紹了autograd,nn包依賴於autograd包來定義模型並對它們求導。乙個nn.module包含各個層和乙個forward input 方法,該方法返回output。例如,下面這個神經網路可以對數...

PyTorch 1 0 中文文件 資料型別資訊

譯者 馮寶寶 可以通過torch.finfo或torch.iinfo訪問torch.dtype的數字屬性。class torch finfotorch.finfo是乙個用來表示浮點torch.dtype的數字屬性的物件 即torch.float32,torch.float64和torch.float...

PyTorch 1 0宣布用於研究和生產AI專案

在最近的一篇博文中,臉書的副總裁bill jia宣布了pytorch的新版本1.0的發布。pytorch是用於python的開源ai框架軟體包,具有tensor計算,包括強大的gpu加速和建立於基於磁帶autodiff系統之上的深度神經網路。該新版本的重要性在於,減少了從研究過程到生產過程中移動ai...