mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +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 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| accuracy.py | ||
| logger.py | ||
| loss.py | ||
| monitor.py | ||
| plugin.py | ||
| progress.py | ||
| time.py | ||