Как заставить модуль Pytorch nn работать с сельдереем? - PullRequest
Новая
       28

Как заставить модуль Pytorch nn работать с сельдереем?

1 голос
/ 13 января 2020

Работа с чужим кодом здесь. Я получил от него модель нейронной сети, сохраненную в виде nn-classify-cpu_0.pkl

Я пытаюсь развернуть модель через сельдерей. Для этого я определяю пользовательский класс в своем основном как 

class NN(nn.Module):##Neural network
    def __init__(self):
        super(NN, self).__init__()
        n_layers = len(hidden_layer_sizes)
        .....

Затем загружаю модель как 

torch.load(f='nn-classify-cpu_0.pkl')

Проблема в том, что когда я хочу запустить сервер сельдерея с помощью Приведенный выше класс NN выдает ошибку «1010 * 

Can't get attribute 'NN' on <module 'celery.bin.celery' from '/home/modeladmin/.local/lib/python3.6/site-packages/celery/bin/celery.py'>
* 1012». Почему модель отлично загружается, когда я запускаю отдельную функцию, но не удается, когда я применяю декоратор сельдерея? Это что-то вроде того, что я должен сделать класс NN видимым для сельдерея каким-то особым образом?

Я использую сельдерей 4.3.0 на python 3.6

...