/pytorch_guide

pytorch samples to help student learning quikly

Primary LanguagePython

pytorch tutorial basic

  1. install guide
  2. tensor guide
  3. autograd guide
  4. model_build guide
  5. data guide
  6. transforms guide
  7. save_load guide
  8. model train guide

pytorch tutorial advanced

  1. tensorboard guide
  2. distribute guide
  3. c++ train
  4. model deploy guide
  • flash
  • docker
  • cloud
  • c++

Train model by your self

three step

  1. tutorial
  2. example
  3. well know website

常用的pytorch 模块

import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim  # adam sgd lr_scheduler
import torch.utils.data as data # dataset dataloader
import torch.optim.lr_scheduler import as lr_scheduler
from torch.utils.tensorboard import SummaryWriter
from torchvision import datasets, transforms
improt torch.jit as jit