Installation/ Docs
Pytorch
Flax
Libtorch
Layers
Loading
Model
Number of epoch
Batch size
Learning rate
Loss functions
Optimizers
Learning rate scheduler
Selected Layers
Model Layout
Code
Model
Epochs : 25
Batch size : 32
Learning rate : 0.01