Fix deprecation warning in sigmoid

This commit is contained in:
Johannes Fischer
2021-07-20 12:52:02 +02:00
parent a9c6857b5b
commit ee7b6b607f
2 changed files with 8 additions and 6 deletions

View File

@@ -2,13 +2,11 @@ import torch
from torch.nn import functional
def parse_functional(functional_config):
if functional_config is None:
return None
elif isinstance(functional_config, str):
if isinstance(functional_config, str):
if functional_config == 'relu':
return functional.relu
elif functional_config == 'sigmoid':
return functional.sigmoid
return torch.sigmoid
elif functional_config == 'softmax':
return functional.softmax
return None