mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
Summary: The loss plugin was using the old-style loss[0] access, which in PyTorch 0.4 and later is an attempt to index into a scalar, generating a warning. Replaced that with loss.item(). This fixes https://github.com/pytorch/pytorch/issues/9142 Closes https://github.com/pytorch/pytorch/pull/9143 Differential Revision: D8726403 Pulled By: ezyang fbshipit-source-id: 6c496b140a74d22c8423f511db901b18615fd6fa |
||
|---|---|---|
| .. | ||
| plugins | ||
| __init__.py | ||
| trainer.py | ||