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

Make drop_prob mutable in DropBlock ScriptModules #28

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

Make drop_prob mutable in DropBlock ScriptModules #28

wants to merge 1 commit into from

Commits on Feb 5, 2019

  1. Make drop_prob mutable in DropBlock ScriptModules 

    This makes it possible to schedule drop_prob during training when used
in JIT mode. The LinearScheduler module still can't be used in JIT
mode yet, but you can change drop_prob from outside of the model, i.e. in
your training loop.
    @mdraw
    mdraw committed Feb 5, 2019
    Configuration menu
    Copy the full SHA
    b1dc305 View commit details
    Browse the repository at this point in the history