disable checkpoints

[ipekoztas]

Hi,

Thank you for your great work. I am doing some experiments and I wonder if there is a way to disable the checkpoints for various dimension checks. I am playing with the dimensions and I struggle with getting warnings and errors. I tried to change the defined constants but I could not manage to change all of them.

I would appreciate your help.

[ZexinHe]

Hi,

Thanks for asking. There are indeed various assertions written to ensure the integrity of the codebase. Although a few of them are magical numbers, such as the Triplane dimension of 3, most of the other assertions are general and should be kept no matter how you change the dimensions. Therefore, I would suggest manually disabling the ones that you do not need FOR SURE.

Bests

[ipekoztas]

Hi,

Thank you so much for your answer. My main goal is to double the size of the image features passing through the transformer-decoder. I am updating the size (32x32)x768 to (32x32)x1536 by some feature concatenation. Do you suggest disabling the assertions will solve the problem? Also, in which file are they defined exactly?

Thank you again for your time.

[ZexinHe]

Since you are simply changing the dimensions of the token, I guess there shouldn't be much assertions to disable. I'd suggest debugging after your modifications and see if any assertions are triggered, and then try to disable them based on your actual intention.

Bests