Skip to content

Commit 8de8e5d

Browse files
authored
Update README.md for pytorch 0.4
1 parent 1c861f9 commit 8de8e5d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,9 @@ I recommend to write your own dataloader using [torch.utils.data.Dataset](http:/
2525
since `multiprocessing.Pool.imap` won't stop even there is no enough memory space.
2626
An example of `dataloader` for VOCDataset: [issue71](https://github.com/longcw/yolo2-pytorch/issues/71).
2727

28+
**NOTE 3:**
29+
Upgrade to PyTorch 0.4: https://github.com/longcw/yolo2-pytorch/issues/59
30+
2831

2932

3033
## Installation and demo

0 commit comments

Comments
 (0)