3 #########################################################################
4 # This program is free software: you can redistribute it and/or modify #
5 # it under the terms of the version 3 of the GNU General Public License #
6 # as published by the Free Software Foundation. #
8 # This program is distributed in the hope that it will be useful, but #
9 # WITHOUT ANY WARRANTY; without even the implied warranty of #
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU #
11 # General Public License for more details. #
13 # You should have received a copy of the GNU General Public License #
14 # along with this program. If not, see <http://www.gnu.org/licenses/>. #
16 # Written by and Copyright (C) Francois Fleuret #
17 # Contact <francois.fleuret@idiap.ch> for comments & bug reports #
18 #########################################################################
23 from torch.nn import functional as fn
24 from torch import Tensor
25 from torch.nn import Module
30 def __init__(self, input_dim, hidden_dim, output_dim):
31 super(MLP, self).__init__()
32 self.fc1 = nn.Linear(input_dim, hidden_dim)
33 self.fc2 = nn.Linear(hidden_dim, output_dim)
42 input = Tensor(100, 10).normal_()
43 target = Tensor(100, 1).normal_()
45 criterion = nn.MSELoss()
46 loss = criterion(output, target)
48 agtree2dot.save_dot(loss,
53 mlp.fc1.weight: 'weight1',
54 mlp.fc1.bias: 'bias1',
55 mlp.fc2.weight: 'weight2',
56 mlp.fc2.bias: 'bias2',
58 open('./mlp.dot', 'w'))
60 print('Generated mlp.dot')
64 fontname = 'Computer Modern'
66 subprocess.check_call(['dot', 'mlp.dot',
69 '-Efontname=' + fontname, '-Efontsize=' + str(fontsize),
70 '-Nfontname=' + fontname, '-Nfontsize=' + str(fontsize),
73 except subprocess.CalledProcessError:
75 print('Calling the dot command failed. Is Graphviz installed?')
78 print('Generated mlp.pdf')