Basics of (2d) convolution (D2L 6.2) | PyTorch | conv2d_torch.ipynb |
CIFAR image classification with CNN | PyTorch | cnn_cifar_pytorch.ipynb |
LeNet (D2L 6.5) | PyTorch | lenet_torch.ipynb |
Batchnorm (D2L 7.5 | PyTorch | batchnorm_torch.ipynb |
Residual networks (D2L 7.6) | PyTorch | resnet_torch.ipynb |
Resnets for CIFAR10 | PyTorch Lightning | cifar10_cnn_lightning.ipynb |
Dense nets (D2L 7.7) | PyTorch | densenet_torch.ipynb |
Multi-GPU training (D2L 12.5) | PyTorch | d2lbook |
Image augmentation (D2L 13.1) | PyTorch | image_augmentation_torch.ipynb |
Imagenet classifier | JAX/Flax | External |
MNIST image classification using a CNN | TF2 | cnn_mnist_tf.ipynb |
Transfer learning for flower classification | TF2 | External |
Deep dream | TF2 | External |
Neural style transfer | TF2 | External) |