Developer Tips: Usage

Loss functions

When using the default nn.CrossEntropyLoss() you may not also specify a final prediction layer activation. As specified in the PyTorch docs nn.CrossEntropyLoss() combines nn.LogSoftmax() and nn.NLLLoss() into one single class. Vulcan is designed to automatically apply nn.Softmax() to all predictions when using this criterion.