_is_ddp = get_rank() > -1 def update(self, outputs: torch. ... Tensor = torch.sum(torch.vstack(value), dim=0).to(device) statistics[key] = value for ...
確定! 回上一頁