diff --git a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py index 35b247e3128b0db44673e021c1203666ae2cbd16..ca97b9ba21e27dfcd3629f2d2bac240430bfb454 100644 --- a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py +++ b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py @@ -470,7 +470,7 @@ class StreamingFCLayer_Batch(HLSCustomOp): """Saves weights into .npy file""" np.save( - os.path.join(code_gen_dir, "weights.npy"), weight_tensor, + os.path.join(code_gen_dir, "weights.npy"), weight_tensor_flipped, ) """Saves weights into .dat file"""