remove PyTorch 1.5 builds

Summary: PyTorch 1.6.0 came out on 28 Jul 2020. Stop builds for 1.5.0 and 1.5.1. Also update the news section of the README for recent releases.

Reviewed By: nikhilaravi

Differential Revision: D31442830

fbshipit-source-id: 20bdd8a07090776d0461240e71c6536d874615f6
This commit is contained in:
Jeremy Reizenstein
2021-10-11 06:12:06 -07:00
committed by Facebook GitHub Bot
parent 6d36c1e2b0
commit 53d99671bd
9 changed files with 8 additions and 140 deletions

View File

@@ -69,12 +69,7 @@ class HarmonicEmbedding(torch.nn.Module):
dtype=torch.float32,
)
try:
self.register_buffer("_frequencies", omega0 * frequencies, persistent=False)
except TypeError:
# workaround for pytorch<1.6
self.register_buffer("_frequencies", omega0 * frequencies)
self.register_buffer("_frequencies", omega0 * frequencies, persistent=False)
self.include_input = include_input
def forward(self, x: torch.Tensor) -> torch.Tensor: