aptlin/cifar-100-densenet

Image classification demo of DenseNet in PyTorch on CIFAR 100

Image classification demo of DenseNet in PyTorch on CIFAR 100

Architecture

Model Architecture

Dense Block

Results

• Comet.ML Logs
• Train F1
• Train accuracy
• Loss
• Test accuracy & F1

References

1. Original Paper: Densely Connected Convolutional Networks
2. Tutorial: Implementing DenseNet from Scratch
3. PyTorch Vision: Reference Implementation