Skip to content

Instantly share code, notes, and snippets.

View motiur's full-sized avatar

Motiur Rahman motiur

  • 13:20 (UTC -12:00)
View GitHub Profile
@kendricktan
kendricktan / capsule_networks.py
Last active August 17, 2021 17:12
Clean Code for Capsule Networks
"""
Dynamic Routing Between Capsules
https://arxiv.org/abs/1710.09829
"""
import torch
import torch.nn as nn
import torch.optim as optim
import torch.nn.functional as F
import torchvision.transforms as transforms