From 9fb2f6cfe67017ea59d48b085f7a3b374e3842c6 Mon Sep 17 00:00:00 2001 From: Csaba Kiraly Date: Tue, 5 Dec 2023 10:37:09 +0100 Subject: [PATCH] update example smallConf with N/K Signed-off-by: Csaba Kiraly --- smallConf.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/smallConf.py b/smallConf.py index 3d75446..a166cec 100644 --- a/smallConf.py +++ b/smallConf.py @@ -107,6 +107,7 @@ def nextShape(): # Network Degree has to be an even number if netDegree % 2 == 0: blockSizeR = blockSizeC = blockSize + blockSizeRK = blockSizeCK = blockSize // 2 chiR = chiC = chi - shape = Shape(blockSizeR, blockSizeC, nn, fm, fr, class1ratio, chiR, chiC, vpn1, vpn2, netDegree, bwUplinkProd, bwUplink1, bwUplink2, run) + shape = Shape(blockSizeR, blockSizeRK, blockSizeC, blockSizeCK, nn, fm, fr, class1ratio, chiR, chiC, vpn1, vpn2, netDegree, bwUplinkProd, bwUplink1, bwUplink2, run) yield shape