byol-pytorch icon indicating copy to clipboard operation
byol-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon lucidrains

CKPT load

Open ansleepykid opened this issue 4 years ago • 1 comments

How can we load the CKPT to get the online model

ansleepykid avatar Nov 13 '20 04:11 ansleepykid

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

lucidrains avatar Nov 13 '20 20:11 lucidrains