[DL] Deep MIL

Deep Multiple Instance Learning

• reference: https://github.com/Roc-Ng/DeepMIL

Error 해결 과정

• 참고한 코드에서는 Visdom이라는 시각화 도구 사용

  • colab에서 Visdom을 사용할 수 없어서 Tensorboard를 사용하는 것으로 코드 수정

• RuntimeError: Found no NVIDIA driver on your system.

  • colab에서 cpu로 실행하면 발생했다. gpu로 바꿨더니 해결

• RuntimeError: mat1 and mat2 shapes cannot be multiplied (505x1024 and 4096x512)

  • feature size를 4096에서 1024로 수정

• RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0!

    pred = torch.zeros(0)  # 변경 전
    pred = torch.zeros(0).to(device)  # 변경 후

• ValueError: Found input variables with inconsistent numbers of samples: [674224, 674368]
  • gt: 674224, pred:674368
  • 코드 다시 확인해서 크기 맞추기