5,337
edits
No edit summary |
|||
Line 10: | Line 10: | ||
==Usage== | ==Usage== | ||
==Memory Usage== | |||
Reducing memory usage | |||
* Save loss using [https://pytorch.org/docs/stable/tensors.html#torch.Tensor.item <code>.item()</code>] which returns a standard Python number |