ctx.has_running_stats = running_mean is not None and running_mean is not None

mapillary / inplace_abn

In-Place Activated BatchNorm for Memory-Optimized Training of DNNs

BSD 3-Clause "New" or "Revised" License

1.32k stars 187 forks source link

Open zfy-csu opened 3 years ago

zfy-csu commented 3 years ago

Maybe it should be ctx.has_running_stats = running_mean is not None and running_var is not None