Goto

Collaborating Authors

 large-scaleclassificati...


GPU-AcceleratedPrimalLearningforExtremelyFast Large-ScaleClassification: SupplementaryMaterial

Neural Information Processing Systems

A binary logistic regression classifier was implemented inPyTorch (v1.4.0 ) and trained over the rcv1 dataset to illustrate the speed ups possible using a GPU (Nvidia Tesla V100) versus only multithreading (24 CPU threads using an Intel Xeon Gold 5118). Speedups were tested for both batch gradient descent (with a 0.001 learning rate) andL-BFGS.