Fix issue #452 for PyTorch 0.4.0

03f5f8c4 · Benedict · 81c9033a · 03f5f8c4
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

lib/roi_data_layer/roibatchLoader.py lib/roi_data_layer/roibatchLoader.py +1 -1

未找到文件。
--- a/lib/roi_data_layer/roibatchLoader.py
+++ b/lib/roi_data_layer/roibatchLoader.py
@@ -51,7 +51,7 @@ class roibatchLoader(data.Dataset):
            # for ratio cross 1, we make it to be 1.
            target_ratio = 1

-        self.ratio_list_batch[left_idx:(right_idx+1)] = target_ratio
+        self.ratio_list_batch[left_idx:(right_idx+1)] = torch.tensor(target_ratio.astype(np.float64)) # trainset ratio list ,each batch is same number


  def __getitem__(self, index):