Starkguava

Is it possible to use tensorflow's own weight pruning api for sparse training and pruning

js error

3

tfjs:formatted:7997 Uncaught (in promise) Error: Batch normalization gradient requires mean and offset to have equal ranks. at lv (tfjs:formatted:7997) at batchNorm_ (tfjs:formatted:13935) at Object.batchNorm__op [as batchNorm] (tfjs:formatted:10410) at batchnorm (pix2pix.js:34)...