Как сделать мой класс Binarizer дружественным к рассолу? (наследование torch.autograd.function) ⇐ Python

[Anonymous]

У меня есть класс, который я использую внутри своей RNN для бинаризации серии масок, которые я изучаю на своем GRU. Однако когда я пытаюсь сохранить свои контрольные точки с помощью Pickle, я всегда получаю сообщение об ошибке, сообщающее, что этот объект невозможно замариновать. Кто-нибудь знает, в чем проблема?
class Binarizer(torch.autograd.Function):
"""Binarizes {0, 1} a real-valued tensor."""

@staticmethod
def forward(ctx, inputs, threshold=5e-3):
outputs = inputs.clone()
outputs[inputs threshold] = 1
return outputs

@staticmethod
def backward(ctx, grad_output):
return grad_output, None


Подробнее здесь: https://stackoverflow.com/questions/791 ... grad-funct