Update train.py

train.py

@@ -88,7 +88,7 @@ def train(dataloader, model, loss_fn, optimizer):
         X, y = X.to(device), y.to(device)
               
         pred = model(X)
-        loss = loss_fn(pred,y)
+        loss = loss_fn(pred, y)
                
         optimizer.zero_grad()
         loss.backward()