ComputerVision Jack

Exception in thread Thread -3: 해당 오류는 Classification Task 진행하던 도중에 만나게 되었다. pytorch lightning module 사용하여 Trainer 정의할 때, worker 설정과 관련한 오류로, worker 맞게 설정하거나 낮추면 해결 가능하다. trainer = pl.Trainer( max_epochs=cfg['epochs'], logger=TensorBoardLogger(cfg['save_dir'], make_model_name(cfg)), gpus=cfg['gpus'], accelerator='ddp' if platform.system() != 'Windows' else None, plugins=DDPPlugin() if platform.sy..

TypeError: Conv2d() received an incalid combination of arguments 해당 오류는 Classification Task 진행하던 도중에 만나게 되었다. Module에 관련된 오류로 Conv2d 자체의 문제 보단 input으로 들어가는 데이터와 관련된 오류이다. # 이미지 가시화 진행 return {'img': transformed, 'class': label} # 학습 적용시 return transformed, label 따라서 visualize 진행할 경우 dict 자료 구조로 반환을 넘겨주면 되지만 학습시엔 label 자체를 넘겨줘야한다.