Ask Your Question

Revision history [back]

click to hide/show revision 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.