Skip to content
This repository has been archived by the owner on Aug 18, 2021. It is now read-only.

X entropy loss input dimensions #123

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
Open

X entropy loss input dimensions #123

wants to merge 2 commits into from

Commits on Dec 18, 2018

  1. Fix CrossEntropyLoss input dimension

    @rubencart
    rubencart committed Dec 18, 2018
    Configuration menu
    Copy the full SHA
    7facb8c View commit details
    Browse the repository at this point in the history

  2. Use .item() to acces element in 0-dim tensor

    @rubencart
    rubencart committed Dec 18, 2018
    Configuration menu
    Copy the full SHA
    dcf1c21 View commit details
    Browse the repository at this point in the history