feiwang3311 / Lantern

BSD 3-Clause "New" or "Revised" License
167 stars 15 forks source link

Implement GPU nllLoss and array transfer ops. #40

Closed dan-zheng closed 5 years ago

dan-zheng commented 5 years ago

GPU nllLoss and nllLoss_grad behave as expected (verified in lantern-cudnn-nll-loss). MNIST CNN works using GPU nllLoss.

dan-zheng commented 5 years ago

Ready for review. There are two main changes:

MNIST CNN works using GPU nllLoss.