Hi!, I was able to find the issue and rectified it in my case, you are right the problem is not with pytorch transforms. Let me summarise it for your understanding.
Lets say I create a ImageFolder, My data is strictly in the order PyTorch expects it t be in case of ImageFolder.
data = ImageFolder(root='/content/flower_data/train',transform=transform)

Here Transforms are from Albumentations Library,
transform = A.Compose([
    A.RandomCrop(width=256, height=256))],ToTensorV2()]) 

Then I end up with this issue,
TypeError: __call__() takes 1 positional argument but 2 were given 
The Code works fine with pytorch transforms, when i create my own custom dataset class like i mentioned in one of my above answers, then the problem with albumntaions library is also solved . I checked their issues(Albumentations) on GitHub, turns out few more people had this issue, there was no conclusive solution, At the moment the solution is to create your own Dataset Function and then use their library, it is same like regular PyTorch transforms or use ImageFolder Dataset with PyTorch transforms. Hope this clarifies it. Cheers!