Просмотр исходного кода

torch.from_tensor() bug fix

5.0
Glenn Jocher 4 лет назад
Родитель
Сommit
09402a2174
1 измененных файлов: 1 добавлений и 1 удалений
  1. +1
    -1
      train.py

+ 1
- 1
train.py Просмотреть файл

@@ -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()

Загрузка…
Отмена
Сохранить