Spaces:
Build error
Build error
PeteBleackley
commited on
Commit
·
a8c528d
1
Parent(s):
e556cb6
Trainable => trainable
Browse files
qarac/models/layers/GlobalAttentionPoolingHead.py
CHANGED
|
@@ -49,7 +49,7 @@ class GlobalAttentionPoolingHead(keras.layers.Layer):
|
|
| 49 |
width = input_shape[-1]
|
| 50 |
self.global_projection = self.add_weight('global projection',
|
| 51 |
shape=(width,width),
|
| 52 |
-
|
| 53 |
self.local_projection = self.add_weight('local projection',
|
| 54 |
shape=(width,width),
|
| 55 |
trainable=True)
|
|
|
|
| 49 |
width = input_shape[-1]
|
| 50 |
self.global_projection = self.add_weight('global projection',
|
| 51 |
shape=(width,width),
|
| 52 |
+
trainable=True)
|
| 53 |
self.local_projection = self.add_weight('local projection',
|
| 54 |
shape=(width,width),
|
| 55 |
trainable=True)
|