Skip to content

[WIP] torch.jit support#27

Open
miguelvr wants to merge 1 commit intomasterfrom
torchscript
Open

[WIP] torch.jit support#27
miguelvr wants to merge 1 commit intomasterfrom
torchscript

Conversation

@miguelvr
Copy link
Copy Markdown
Owner

@miguelvr miguelvr commented Feb 3, 2019

Refactor of DropBlock2D and DropBlock3D in TorchScript, while maintaining backwards compatibility.

Fixes #25

To Do:

  • convert DropBlock2D
  • convert DropBlock3D
  • convert LinearScheduler

@mdraw mdraw mentioned this pull request Feb 4, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant