Hi @fahad, it failed at this line
    Map<String, IValue> outTensors = mModule.forward(IValue.from(inputTensor)).toDictStringKey();

The error message said that it expects DICT_STRING_KEY (type 13) but got TUPLE (type 7). That means your model returns a tuple instead of dict. You need to call toTuple() instead.
    IValue[] outputTuple = mModule.forward(IValue.from(inputTensor)).toTuple();