Skip to content

Pytorch

0 Quickstart

  • nn.CrossEntropy ???
  • argmax ???

1 Tensor

  • Squeeze ???
  • Unsqueeze ???

Dataset and Dataloader

  • torch.randint(size=(1,)) ???

Auto Gradient

Basics

  • Dataset and DataLoaders
  • Transforms
  • Model
  • Optimization
  • Save and Load Model

2.3 torch.nn

  • Network Layers
  • Network Parameters
  • Non-Linear Activation Functions
  • Build your own neural network
  • Loss Functions