From 9d87471579c80d1c8baff6711c1297dec8f0dcf4 Mon Sep 17 00:00:00 2001 From: Benjamin Fiske Date: Wed, 4 May 2022 23:17:41 -0400 Subject: adding vgg weights --- losses.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) (limited to 'losses.py') diff --git a/losses.py b/losses.py index bc241282..c0989ed1 100644 --- a/losses.py +++ b/losses.py @@ -64,15 +64,17 @@ class YourModel(tf.keras.Model): for layer in self.vgg16: layer.trainable = False + self.layer_to_filters = {layer.name: layer.filters for layer in self.vgg16 if "conv" in layer.name} self.indexed_layers = [layer for layer in self.vgg16 if "conv1" in layer.name] self.desired = [layer.name for layer in self.vgg16 if "conv1" in layer.name] + self.vgg16 = tf.keras.Sequential(self.vgg16, name="vgg") + # create a map of the layers to their corresponding number of filters if it is a convolutional layer - self.layer_to_filters = {layer.name: layer.filters for layer in self.vgg16 if "conv" in layer.name} def call(self, x): layers = [] - for layer in self.vgg16: + for layer in self.vgg16.layers: # pass the x through x = layer(x) # print("Sotech117 is so so sus") @@ -159,6 +161,3 @@ class YourModel(tf.keras.Model): print(type(self.x)) print(type(gradients)) self.optimizer.apply_gradients(zip(gradients, [self.x])) - - - -- cgit v1.2.3-70-g09d2