Commit aafe3640 authored by Mira Arabi Haddad's avatar Mira Arabi Haddad
Browse files

GROUPWORK_Refactoring

parent 3925a8a4
from os import error
# import dlc_practical_prologue as prologue
import torch
from torch import nn
from torch.nn import functional as F
......@@ -8,7 +7,6 @@ from torch.nn.modules.batchnorm import BatchNorm2d
from torch.nn.modules.conv import Conv2d
from torch.nn.modules.dropout import Dropout
from torch.nn.modules.pooling import MaxPool2d
# import matplotlib.pyplot as plt
## Basic MLP Network with 2 hidden layers
class NN(nn.Module):
......@@ -127,6 +125,7 @@ class CNN_Classification(nn.Module):
x2 = x2.view(-1, 576)
x1 = self.classifier(x1)
x2 = self.classifier(x2)
# concatenate both outputs into one tensor of size 20
return torch.cat((x1, x2), 1)
## MLP Network with 2 hidden layers for comparison only
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment