コード例 #1
0
ファイル: detector.py プロジェクト: mseeger/autogluon-1
 def save(self, checkpoint):
     """save object detector to the file specified by 'checkpoint'
     """
     state_dict = self.state_dict()
     save(state_dict, checkpoint)
コード例 #2
0
 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)