Update README.md

2023-05-26 23:06:31 -04:00 · 2023-05-26 23:06:31 -04:00 · 29bb16da2a
parent 0cc458e237
commit 29bb16da2a
1 changed files with 2 additions and 0 deletions
--- a/README.md
+++ b/README.md
@ -364,6 +364,7 @@ Fabric is designed for the most complex models like foundation model scaling, LL
 -         loss.backward()
 +         fabric.backward(loss)
          optimizer.step()
+          print(loss.data)
 ```

 </sub>
@ -397,6 +398,7 @@ for epoch in range(num_epochs):
        loss = torch.nn.functional.cross_entropy(outputs, labels)
        fabric.backward(loss)
        optimizer.step()
+        print(loss.data)
 ```

 </sub>