Eylon Stroh

To address the first issue, try changing ``` squeeze_indices = [1] # Squeeze biases. if node.kind == NodeKind.InnerProduct: squeeze_indices.append(0) # Squeeze FC. ``` to ``` squeeze_indices = [] if node.kind...