diff --git a/mpl-xor/main.py b/mpl-xor/main.py index 7350ed475f6c06b0cf715deb97f05923874d8a7b..7d93ddf4b2ef1ce11232c4d71b0b92433d5cefa9 100644 --- a/mpl-xor/main.py +++ b/mpl-xor/main.py @@ -1,6 +1,5 @@ import torch import torch.nn as nn -import numpy as np train_data = torch.tensor([[0, 0], [0, 1], [1, 0], [1, 1]], dtype=torch.float).cuda() train_data_y = torch.tensor([[0], [1], [1], [1]], dtype=torch.float).cuda() diff --git a/perceptron-or/pytorch.py b/perceptron-or/pytorch.py index 67b59addd26d99d581eda30a6eb4a4099e671c57..5d6900a42a04eabe1ac4b76c624687864f9dce71 100644 --- a/perceptron-or/pytorch.py +++ b/perceptron-or/pytorch.py @@ -1,6 +1,5 @@ import torch import torch.nn as nn -import numpy as np train_data = torch.tensor([[0, 0], [0, 1], [1, 0], [1, 1]], dtype=torch.float).cuda() train_data_y = torch.tensor([[0], [1], [1], [1]], dtype=torch.float).cuda()