This is a PyTorch implementation of DenseNet.
This code has been used for the paper "Visual Explanation of a Deep Learning Solar Flare Forecast Model and Its Relationship to Physical Parameters" (https://doi.org/10.3847/1538-4357/abdebe).
from DenseNet_model import DenseNet
model = DenseNet(layers_num=5, growth_rate=13, drop_rate=0.5).to(device=DEVICE)
transforms.Compose([
transforms.ToTensor(),
transforms.Normalize(
mean=[0.5, 0.5, 0.5],
std=[0.5, 0.5, 0.5])
])