GateNLP / gate-lf-pytorch-json

PyTorch wrapper for the LearningFramework GATE plugin
Apache License 2.0
1 stars 2 forks source link

Allow to include regularization terms in the loss function #22

Open johann-petrak opened 6 years ago

johann-petrak commented 6 years ago

This means the loss function depends, like the optimizer, on potentially all model parameters, so we have to construct it at the same point.

Initially just make it work like this: