5 Commits

Author	SHA1	Message	Date
Fangjun Kuang	4319a187b3	Zero out the gradient of decoder/joiner for auxiliary losses.	2022-02-22 19:44:01 +08:00
Fangjun Kuang	76632bddfe	Minor fixes.	2022-02-22 17:37:14 +08:00
Fangjun Kuang	8b4c571824	Add auxiliary losses.	2022-02-22 16:15:00 +08:00
Fangjun Kuang	5e4b1e01fe	Get the input for the auxiliary branch.	2022-02-21 19:56:45 +08:00
Fangjun Kuang	218d79aba7	Copy files.	2022-02-21 19:32:26 +08:00