[Mask DINO] Train with Custom Dataset

[Error]

1.

File "/home/.../detectron2/detectron2/engine/train_loop.py"

TypeError: __init__() got an unexpected keyword argument 'dtype'

 

1- solution

/detectron2/detectron2/engine/train_loop.py 에서

 

with autocast(dtype=self.precision):

 

위 문장 주석처리

 

2.

UnboundLocalError: local variable 'losses' referenced before assignment

 

2-solution

/detectron2/detectron2/engine/train_loop.py 에서

 

zero_grad_before_forward=False --> True로 변경

또는

 

losses, loss_dict과 같은 parameter들이 하위에 선언되어 있는 것을 상위에 선언하는 형식으로 변경해줌

 

3.

RuntimeWarning: invalid value encountered in intersection

 

3-solution

warnings.filterwarnings('ignore', category=RuntimeWarning)