utils里内容改成
if scale_each is True:
for idx, _ in enumerate([jj for jj in tensor]):
t = tensor[idx]
# for t in tensor: # loop over mini-batch dimension
# norm_range(t, range)
utils里内容改成
if scale_each is True:
for idx, _ in enumerate([jj for jj in tensor]):
t = tensor[idx]
# for t in tensor: # loop over mini-batch dimension
# norm_range(t, range)
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。