Meshgrid indexing='ij' for PyTorch 1.10 (#5309)
* Meshgrid `indexing='ij'` for PyTorch 1.10 Will not merge currently as breaks backwards compatibility. * Meshgrid `indexing='ij'` for PyTorch 1.10 Will not merge currently as breaks backwards compatibility. * Add check_version hard argument * Update comment
This commit is contained in:
1 parent
5d4258fac5
commit
8c326a1edf
3 files changed
+13
-7
No files matched your search
@@ -20,7 +20,7 @@ class Albumentations:
|
||||
self.transform = None
|
||||
try:
|
||||
import albumentations as A
|
||||
check_version(A.__version__, '1.0.3') # version requirement
|
||||
check_version(A.__version__, '1.0.3', hard=True) # version requirement
|
||||
|
||||
self.transform = A.Compose([
|
||||
A.Blur(p=0.01),
|
||||
|
||||
Reference in new issue
Block a user