Сериализация полей классов в моделях PyTorch

Question

Существует ли простой способ сериализации не только параметров модели в PyTorch? Например, список целых чисел?

Позвольте мне объяснить. Скажем, у нас очень простая сеть PyTorch:

import torch

class Net(torch.nn.Module):

    some_integers = []

    def __init__(self):
        # stuff

    def forward(self, x):
        # more stuff

Мы можем использовать torch.save() для сериализации (сохранения в файл) параметров обученной сети, но для класса выше он не сохранит поле some_integers .

Я знаю, что мы могли бы использовать Pickle или что-то подобное для сериализации поля, но тогда я бы получил два файла, что нежелательно в моем случае.