Solved by Alberto_Albiol in post #3 
              
              
                I faced the same issue, the solution is simple: 
   sample = {'image':image, 'label':label}

    if self.transform:
        sample = self.transform(**sample)

notice the ** in self.transform