Don't reshape empty weights

klarh · Mar 4, 2021 · 5a8fad8 · 5a8fad8
1 parent 3aa6135
commit 5a8fad8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/keras_gtar/trajectory.py b/keras_gtar/trajectory.py
@@ -85,7 +85,8 @@ def get_weights(self, frame=-1):
                 shape_rec = shape_records[i][weight_index]
                 shape = self.handle.getRecord(shape_rec, frame_index)
                 weight = self.handle.getRecord(weight_rec, frame_index)
-                weight = weight.reshape(shape)
+                if weight.size:
+                    weight = weight.reshape(shape)
                 all_weights.append(weight)
 
         return all_weights