GateNLP / gate-lf-pytorch-json

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

Always use CPU for application for now #40

Closed johann-petrak closed 5 years ago

johann-petrak commented 5 years ago

The learning framework sends over individual instances for application. The main reason for this is that this is necessary if we want to use the special feature that gives us the classification of the previous instance.

We could restrict the batch size to 1 only if we actually use that feature, but for now we stick with batch size 1.

In that case, using the GPU is likely to be slower because of the additional transfer overhead, so for now, lets always use the CPU.

johann-petrak commented 5 years ago

This should now work.

johann-petrak commented 5 years ago

Works