File 06_2_model_loss_and_ optimizer.py

06_2_model_loss_and_ optimizer.py:None–None

# 1) Design model (input, output, forward pass with different layers)
# 2) Construct loss and optimizer
# 3) Training loop
#       - Forward = compute prediction and loss