Fork me on CSDN
2021年12月30日
Batch Normalization Code
摘要： def batch_norm(X, gamma, beta, moving_mean, moving_var, eps, momentum): if not torch.is_grad_enabled():#In prediction mode X_hat = (X-moving_mean)/tor 阅读全文
posted @ 2021-12-30 15:25 追风赶月的少年 阅读(51) 评论(0) 推荐(0) 编辑