Nguyễn Trung Kiên

I try to load checkpoint by using argument callbacks but I get that problem Can you show me another way to load the checkpoint work fine. Thank you!

I can't see you use patch_size in your implement. Can you explain for this issue?