Предупреждение Pytorch отображается несмотря на фильтрацию - PullRequest
0 голосов
/ 29 марта 2020

Я подгоняю набор данных, используя библиотеку fastai. Код, вызывающий проблему, находится здесь:

learn = language_model_learner(data_lm, TransformerXL, path='training_data')

with warnings.catch_warnings():

    warnings.simplefilter("ignore")
    warnings.filterwarnings("ignore")

    learn.fit_one_cycle(epochs, 1e-2)

Он выдает предупреждение каждый шаг, несмотря на фильтр:

Warning: masked_fill_ received a mask with dtype torch.uint8, this behavior is now deprecated,please use a mask with dtype torch.bool instead. (masked_fill__cuda at /pytorch/aten/src/ATen/native/cuda/LegacyDefinitions.cpp:19)

Код работает, и я не могу Не используйте Pytorch напрямую, поэтому я бы предпочел проигнорировать предупреждение. В частности, предупреждение заполняет консоль, предотвращая отображение статистики потерь эпохи. Похоже, что он обрабатывается стандартной библиотекой, так почему же фильтр не работает? Я пробовал с и без защелки и оба фильтра.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...