예제 #1
0
 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)
예제 #3
0
 def save(self, checkpoint):
     """ Save image classifier to folder specified by `checkpoint`.
     """
     state_dict = self.state_dict()
     save(state_dict, checkpoint)