torch-toolbox
torch-toolbox copied to clipboard
🛠 Toolbox to extend PyTorch functionalities
Thanks for your repo. Have you considered supporting 3D convolution?
Hi, I am new to the package. I want to use the ImageNetPolicy as one of the data transformation by `from torchtoolbox.transform import ImageNetPolicy` But shows: ImportError: cannot import name...
DCNv2
How exactly is DCNv2 supported in this repo?
I've looked at the code for evonorm layers, it seems that you expect a 4d (n,c,h,w) shape. I need evonorm with 1d convolutions, for this it seems the only way...
Hello, when I use cutout, the following error occurs: from torchtoolbox.transform import Cutout File "/anaconda3/envs/recognition/lib/python3.8/site-packages/torchtoolbox/transform/init.py", line 4, in from .transforms import * File "/anaconda3/envs/recognition/lib/python3.8/site-packages/torchtoolbox/transform/transforms.py", line 11, in from . import...
https://github.com/PistonY/torch-toolbox/blame/993a0dfe56de9eda8d12aa08ebdae6ee932545cf/torchtoolbox/transform/autoaugment.py#L104 For the above line, following error is encountered ``` /usr/local/lib/python3.10/dist-packages/torchtoolbox/transform/autoaugment.py in __init__(self, p, magnitude) 102 class Posterize(SubPolicy): 103 def __init__(self, p, magnitude=None): --> 104 ranges = np.round(np.linspace(8, 4, 10),...