Glenn Jocher
4 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
train.py
View File
| @@ -225,7 +225,7 @@ def train(hyp, opt, device, tb_writer=None): | |||
| if rank != -1: | |||
| indices = torch.zeros([dataset.n], dtype=torch.int) | |||
| if rank == 0: | |||
| indices[:] = torch.from_tensor(dataset.indices, dtype=torch.int) | |||
| indices[:] = torch.tensor(dataset.indices, dtype=torch.int) | |||
| dist.broadcast(indices, 0) | |||
| if rank != 0: | |||
| dataset.indices = indices.cpu().numpy() | |||
Loading…