search

Qihoo360 / tensornet

Apache License 2.0
315 stars 72 forks source link

tensornet 支持batchnorm #58

Open welsonzhang opened 3 months ago

welsonzhang commented 3 months ago

batch norm中的moving_mean和moving_variance 参数非trainable的参数，tensornet只支持trainable参数，可以支持一下batch norm实现吗？