[이왜안?] PicklingError: Can't pickle <function <lambda> at 0x000002AC976E6CA0>: attribute lookup <lambda> on __main__ failed

전혀 감도 안오는 에러가 뜨길래 약간 쫄았는데 나도 짬이 얼추 찼나보다. 금방 해?결 했다.

torch의 DataLoader를 설정하는 부분에서 num_workers를 0이 아닌 다른 값을 전달했다면 0으로 바꿔보고 다시 실행해보자.

 

test_loader = DataLoader(
    test_set,
    batch_sampler=test_sampler,
    num_workers=0, # 이 값을 0 이외의 숫자로 전달하지 않았는지 확인하자.
    pin_memory=True,
    collate_fn=test_sampler.episodic_collate_fn,
)