Skip to content

Commit

Permalink
remove gradcheck import for torch<1.7
Browse files Browse the repository at this point in the history
  • Loading branch information
chrischoy committed Dec 15, 2020
1 parent 2039267 commit cab49b4
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 1 deletion.
1 change: 0 additions & 1 deletion MinkowskiEngine/utils/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,5 @@
# of the code.
from .quantization import sparse_quantize, ravel_hash_vec, fnv_hash_vec
from .collation import SparseCollation, batched_coordinates, sparse_collate, batch_sparse_collate
from .gradcheck import gradcheck
from .coords import get_coords_map
from .init import kaiming_normal_
5 changes: 5 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -301,6 +301,11 @@ For generative transposed convolution, please cite:
}
```


## Unittest

For unittests and gradcheck, use torch >= 1.7

## Projects using Minkowski Engine

Please feel free to update [the wiki page](https://github.com/NVIDIA/MinkowskiEngine/wiki/Usage) to add your projects!
Expand Down

0 comments on commit cab49b4

Please sign in to comment.