/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

0 vote199 views2 answers
Pandas rename columns by position
Python
8 votes173 views2 answers
Loop through list backwards python
Python
2 votes157 views1 answer
Difference between set and list in python
Python
3 votes213 views1 answer
Django import model from another app
Python Django
0 vote171 views1 answer
Python red table from pdf
Python
11 votes188 views2 answers
Find character in python
Python
5 votes209 views1 answer
Python flask query params
Python Flask
0 vote167 views1 answer
How to increment date by one in python
Python
16 votes162 views6 answers
Turtle with python
Python
8 votes320 views3 answers
Add new column based on condition on some other column in pandas.
Python