CNN
Datamodule
Evaluation
Utils
Visualization
activations
function
functional
layer
linear
losses
model
net
optimizer
