Dataset

Optimizer

Loss

Training

epochs =
batch_size =
shuffle =