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

Add minor cosmetic changes

Browse Source
This commit is contained in:
blackfly
2020-04-27 12:45:42 +02:00
parent a167565857
commit dab91e471a
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
prototorch/datasets/abstract.py
View File

@@ -1,4 +1,4 @@
"""ProtoTorch abstract datasets
"""ProtoTorch abstract dataset classes.
Based on `torchvision.VisionDataset` and `torchvision.MNIST`
@@ -13,7 +13,7 @@ import torch
class Dataset(torch.utils.data.Dataset):
"""Abstract dataset class to be inherited"""
"""Abstract dataset class to be inherited."""
_repr_indent = 2
def __init__(self, root):
@@ -29,7 +29,7 @@ class Dataset(torch.utils.data.Dataset):
class ProtoDataset(Dataset):
"""Abstract dataset class to be inherited"""
"""Abstract dataset class to be inherited."""
training_file = 'training.pt'
test_file = 'test.pt'

6
prototorch/functions/activations.py
View File

@@ -16,7 +16,7 @@ def register_activation(function):
@register_activation
# @torch.jit.script
def identity(x, beta=torch.tensor([0])):
def identity(x, beta=torch.tensor(0)):
"""Identity activation function.
Definition:
@@ -27,7 +27,7 @@ def identity(x, beta=torch.tensor([0])):
@register_activation
# @torch.jit.script
def sigmoid_beta(x, beta=torch.tensor([10])):
def sigmoid_beta(x, beta=torch.tensor(10)):
r"""Sigmoid activation function with scaling.
Definition:
@@ -42,7 +42,7 @@ def sigmoid_beta(x, beta=torch.tensor([10])):
@register_activation
# @torch.jit.script
def swish_beta(x, beta=torch.tensor([10])):
def swish_beta(x, beta=torch.tensor(10)):
r"""Swish activation function with scaling.
Definition: