mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00: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 |
||
|---|---|---|
| .. | ||
| backcompat | ||
| bottleneck | ||
| data | ||
| ffi | ||
| serialization | ||
| trainer | ||
| __init__.py | ||
| checkpoint.py | ||
| collect_env.py | ||
| cpp_extension.py | ||
| dlpack.py | ||
| file_baton.py | ||
| hooks.py | ||
| model_zoo.py | ||