1 | initial version |
@anomaly_, I'm sorry. Some problems with website and my answers gone.
Firstly, I have recommended you to convert your model to CPU mode by net = net:float()
.
For solve issue with LogSoftMax you could try to remove this layer or replace it to Identity. It's correct because LogSoftMax is a monotonic function. It means that for all A > B: LogSoftMax(A) > LogSoftMax(B). So for model deployment you can just use data from layer before LogSoftMax. Moreover it's better for efficiency.