Epoch 1/10, Loss: 0.30326950550079346
Epoch 2/10, Loss: 0.06523166596889496
Epoch 3/10, Loss: 0.010999132879078388
Epoch 4/10, Loss: 0.027588920667767525
Epoch 5/10, Loss: 0.016694020479917526
Epoch 6/10, Loss: 0.042774200439453125
Epoch 7/10, Loss: 0.00832352228462696
Epoch 8/10, Loss: 0.012629617005586624
Epoch 9/10, Loss: 0.00104866165202111
Epoch 10/10, Loss: 0.000959626107942313
