lucidrains/byol-pytorch

CKPT load

ansleepykid opened this issue · 1 comments

How can we load the CKPT to get the online model

@ansleepykid you can also save the entire BYOL class, and restore the training at another point in time, if that's what you mean