Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add out_ch param to models #10

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Add out_ch param to models #10

wants to merge 1 commit into from
+14 −14

Commits on May 12, 2020

  1. Add out_ch param to models 

    The `out_ch` param was doing nothing!
repaired this.
I would also recommend removing the sigmoid calls at the end, and use BCE with Logits.
    @tcapelle
    tcapelle authored May 12, 2020
    Configuration menu
    Copy the full SHA
    0125c76 View commit details
    Browse the repository at this point in the history