julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[QA] Remove commented-out torch.jit.script decorators

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-06-01 19:46:21 +02:00
parent b1e64c8b8b
commit 2a7394b593
1 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
prototorch/functions/activations.py
View File

@ -5,17 +5,14 @@ import torch
ACTIVATIONS = dict()
# def register_activation(scriptf):
# ACTIVATIONS[scriptf.name] = scriptf
# return scriptf
def register_activation(function):
def register_activation(fn):
"""Add the activation function to the registry."""
ACTIVATIONS[function.__name__] = function
return function
name = fn.__name__
ACTIVATIONS[name] = fn
return fn
@register_activation
# @torch.jit.script
def identity(x, beta=0.0):
"""Identity activation function.
@ -29,7 +26,6 @@ def identity(x, beta=0.0):
@register_activation
# @torch.jit.script
def sigmoid_beta(x, beta=10.0):
r"""Sigmoid activation function with scaling.
@ -44,7 +40,6 @@ def sigmoid_beta(x, beta=10.0):
@register_activation
# @torch.jit.script
def swish_beta(x, beta=10.0):
r"""Swish activation function with scaling.