def save(self, checkpoint): """save object detector to the file specified by 'checkpoint' """ state_dict = self.state_dict() save(state_dict, checkpoint)
def save(self, checkname=None): checkname = checkname if checkname else self.checkname mkdir(os.path.dirname(checkname)) save(self.state_dict(), checkname)
def save(self, checkpoint): """ Save image classifier to folder specified by `checkpoint`. """ state_dict = self.state_dict() save(state_dict, checkpoint)