관리 메뉴

ComputerVision Jack

Exception in thread Thread -3: 본문

Computer Vision/PyTorch Framework ISSUE

Exception in thread Thread -3:

JackYoon 2022. 3. 31. 13:43
반응형
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.system() != 'Windows' else None,
        callbacks=callbacks,
        resume_from_checkpoint=ckpt,
        **cfg['trainer_options'],
    )
trainer.fit(model_module, data_module)
반응형
Comments