приложение cra sh, когда выполнение model.forward на android коде, я не знаю почему? - PullRequest
1 голос
/ 22 апреля 2020

Когда я выполняю этот код точно в строке, которая соответствует module.forward, мои приложения перезапускаются. Код прилагается ниже:

@WorkerThread
@Nullable
public IValue translation(String input){       
     // loading serialized torchscript module from packaged into app android asset model.pt,
     Module module = Module.load(assetFilePath(this.context,"model_seq2seq_22_04_2020_13.pt"));

    inputTensor = generateTensor(lists,input);                 
    Log.d("tensor shape", Arrays.toString(inputTensor.shape()));


    long[] shape = new long[]{1,1};
    long[] start = new long[1];
    start[0]=0;

    Tensor trgStart = Tensor.fromBlob(start,shape);
    Log.d("tensor shape", Arrays.toString(trgStart.shape()));
    IValue output = module.forward(IValue.from(inputTensor),IValue.from(trgStart));    
}
...