Closed f-dangel closed 2 years ago
Support includes SumModule, ScaleModule, and Identity. Fix incorrect detection of reduction factor in the tests if batch_size=1.
SumModule
ScaleModule
Identity
batch_size=1
Support includes
SumModule
,ScaleModule
, andIdentity
. Fix incorrect detection of reduction factor in the tests ifbatch_size=1
.