/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

3 votes339 views2 answers
Draw bounding box on image python cv2
Python
0 vote281 views1 answer
Set page title name and favicon in streamlit
Python
0 vote331 views1 answer
Python Entry default text
Python
0 vote272 views1 answer
Kneighbours regressor sklearn
Python
0 vote271 views1 answer
Python pass command is 'Do Nothing'
Python
0 vote222 views1 answer
Python 3.9 docker install
Python
32 votes295 views6 answers
Remove space from string python
Python
-1 vote292 views1 answer
Flask set mime type
Python
0 vote262 views1 answer
Check pd.NaT python
Python
0 vote277 views1 answer
Random night stars with python turtle
Python