From e049f1d10302cd94a7bb2c08affd359b58f7cefb Mon Sep 17 00:00:00 2001 From: Aravinda Kumar Date: Fri, 12 Jul 2024 19:06:23 +0530 Subject: [PATCH] replaced depracated np.float with np.float64 --- examples/data2vec/models/mae.py | 2 +- fairseq/modules/dynamic_crf_layer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/data2vec/models/mae.py b/examples/data2vec/models/mae.py index a3b5f72a4a..62aa9f8b6b 100644 --- a/examples/data2vec/models/mae.py +++ b/examples/data2vec/models/mae.py @@ -403,7 +403,7 @@ def get_1d_sincos_pos_embed_from_grid(embed_dim, pos): out: (M, D) """ assert embed_dim % 2 == 0 - omega = np.arange(embed_dim // 2, dtype=np.float) + omega = np.arange(embed_dim // 2, dtype=np.float64) omega /= embed_dim / 2.0 omega = 1.0 / 10000 ** omega # (D/2,) diff --git a/fairseq/modules/dynamic_crf_layer.py b/fairseq/modules/dynamic_crf_layer.py index 8fcc6b8d26..b3d8157d66 100644 --- a/fairseq/modules/dynamic_crf_layer.py +++ b/fairseq/modules/dynamic_crf_layer.py @@ -105,7 +105,7 @@ def _compute_normalizer(self, emissions, targets=None, masks=None, beam=None): beam = beam if beam is not None else self.beam batch_size, seq_len = emissions.size()[:2] if targets is not None: - _emissions = emissions.scatter(2, targets[:, :, None], np.float("inf")) + _emissions = emissions.scatter(2, targets[:, :, None], np.float64("inf")) beam_targets = _emissions.topk(beam, 2)[1] beam_emission_scores = emissions.gather(2, beam_targets) else: