/Python/Pytorchvideo cant pickle local object 'kinetics data module.val dataloader.<locals>.<lambda>'

Pytorchvideo cant pickle local object 'kinetics data module.val dataloader.<locals>.<lambda>'

Pytorchvideo cant pickle local object 'Kinetics Data Module.val dataloader.<locals>.<lambda>'

# Replace Lambda function with normal function or reduce
# number of workers to 0

trsfm = transforms.Compose([
  transforms.Resize((32, 32)),
  transforms.ToTensor(),
  transforms.Normalize((0.1307,), (0.3081,)),
  transforms.Lambda(lambda x: x.repeat(3, 1, 1))
])

#Should become

def tmp_func(x):
  return x.repeat(3, 1, 1)

trsfm = transforms.Compose([
  transforms.Resize((32, 32)),
  transforms.ToTensor(),
  transforms.Normalize((0.1307,), (0.3081,)),
  transforms.Lambda(tmp_func)
])

Source: discuss.pytorch.org

Similar codes

2 votes389 views1 answer
Django filter multiple fields
Python Django
0 vote355 views2 answers
Python difference between multiprocessing Pool and Threadpool
Python
0 vote329 views1 answer
Pyqt button hover color
Python
2 votes349 views1 answer
Change false to true python
Python
0 vote355 views4 answers
Python update installed packages
Python
0 vote356 views1 answer
For i in range without i
Python
1 vote354 views3 answers
Opencv grayscale to rgb
Python
4 votes341 views1 answer
Numpy mean 2 arrays
Python
0 vote375 views1 answer
Flask Upload file to local s3
Python Flask
0 vote345 views1 answer
Get derivative of interp1d
Python