Batch 1 이상이면 터짐 about spade_colorization HOT 4 CLOSED

@ThisIsIsaac 혹시 해결된거면 이거는 닫을게 👍

from spade_colorization.

is_reconstructing 변수가 batchsize가 1에서 B로 alter하면서 B일경우는 조건문에 걸리면서 에러를 뱉는다.
언제 alter 되는지, 어떻게 고칠지에 대한 정책 수립 필요

from spade_colorization.

reconstruction 에 관련된 batch error 는 고쳤어. 이젠 한 batch 에 모두 동일하게 reconstruction 이 적용되. 하지만 여전히 #25 이 에러는 있네.

이 코드에 두번째 줄 형이 고친거야?

        if opt.use_reconstruction_loss and i % opt.reconstruction_period == 0:
            data_i["reference_LAB"] = data_i["target_LAB"].clone().detach()
            data_i["is_reconstructing"] = True

from spade_colorization.

잘 기억이 안나네. 아마 그랬던거 같아

from spade_colorization.