projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: ef91cfd) | patch
Loss (#14720)
authorElias Ellison <eellison@fb.com>
Tue, 4 Dec 2018 20:27:22 +0000 (12:27 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 4 Dec 2018 20:30:05 +0000 (12:30 -0800)
commitba70cf22fa867ada25f0dace95868999e53ef91f
treeca66f5825dac74dadb3f70f7101b13f05260d6actree | snapshot
parentef91cfd68b952fc24fd871837be019ced8f76f8ecommit | diff
Loss (#14720)

Summary:
Adding Loss modules to script.  Some of the modules have an optional tensor parameter. I will wait until wanchao's diff to support optional tensors is landed before landing this.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14720

Differential Revision: D13317990

Pulled By: eellison

fbshipit-source-id: 535925bdf126d28d9e7d64077b83ebd836a5beba
test/test_jit.py diff | blob | history
torch/nn/modules/loss.py diff | blob | history
Domain: Machine Learning / ML Framework;