apex icon indicating copy to clipboard operation
apex copied to clipboard

Variant Name Error

Open kk19990709 opened this issue 3 years ago • 0 comments

In https://github.com/NVIDIA/apex/tree/master/apex/parallel

import apex
input_t = torch.randn(3, 5, 20).cuda()
sbn = apex.parallel.SyncBatchNorm(5).cuda()
output_t = sbn(input)

SHOULD BE

import apex
input_t = torch.randn(3, 5, 20).cuda()
sbn = apex.parallel.SyncBatchNorm(5).cuda()
output_t = sbn(input_t)

kk19990709 avatar Sep 15 '22 09:09 kk19990709