Как загрузить модель из факела в пыльник? - PullRequest
Новая
       39

Как загрузить модель из факела в пыльник?

0 голосов
/ 02 марта 2019

Я скачал модели с https://github.com/XingxingZhang/dress.
В каждой папке есть два файла "16.state.t7" (~ 5 МБ) и "16.t7" (~ 100 МБ).Теперь я хочу загрузить модель в pytorch или даже torch и начать вывод.

Когда я загружаю модель "16.state.t7" в pytorch, используя: - 

from torch.utils.serialization import load_lua
model = load_lua('16.state.t7')

И затемdo: - 

type(model)

Я получаю "torch.utils.serialization.read_lua_file.hashable_uniq_dict"

Аналогично, когда я загружаю модель "16.t7", используя: - 

model = load_lua('16.t7')

А затем сделайте: - 

type(model)

Это дает "torch.Tensor"

Также, когда я 

print(load_lua('16.state.t7').keys())

Я получаю: - 

['sgdParam', 'optimMethod', 'dst_vocab', 'validout', 'nhid', 'patience', lmPath', 'sariRevWeight', 'deltaSamplePos', 'initHidVal', 'testout', 'save', 'lrDiv', 'fineTuneFactor', 'useGPU', 'batchSize', 'src_vocab', 'gradClip', 'freqCut', 'savePerEpoch', 'ignoreCase', 'valid', 'lr', 'lmWeight', 'minLR', 'test', 'nivocab', 'learnZ', 'sampleStart', 'rfEpoch', 'novocab', 'validBatchSize', 'nin', 'power', 'sariWeight', 'dropout', 'normalizeUNK', 'encdecPath', 'nneg', 'train', 'initRange', 'recDropout', 'nlayers', 'saveBeforeLrDiv', 'lnZ', 'seqLen', 'curLR', 'minImprovement', 'wordEmbedding', 'seed', 'maxEpoch', 'attention', 'simPath', 'simWeight', 'model', 'embedOption']
...