update example smallConf with N/K
Signed-off-by: Csaba Kiraly <csaba.kiraly@gmail.com>
This commit is contained in:
parent
fbb73265c1
commit
9fb2f6cfe6
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue