diff options
Diffstat (limited to 'losses.py')
| -rw-r--r-- | losses.py | 9 |
1 files changed, 4 insertions, 5 deletions
@@ -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])) - - - |