def save_checkpoint(self, outname='last'):
    '\n        Save all information of the current iteration\n        '
    weights = {'model': self.model.model.state_dict(), 'optimizer': self.optimizer.state_dict(), 'iters': self.iters, 'best_value': self.best_value}
    if (self.scaler is not None):
        weights[self.scaler.state_dict_key] = self.scaler.state_dict()
    self.checkpoint.save(weights, outname)
Save all information of the current iteration
def save_checkpoint(self, outname='last'):
    '\n        \n        '
    weights = {'model': self.model.model.state_dict(), 'optimizer': self.optimizer.state_dict(), 'iters': self.iters, 'best_value': self.best_value}
    if (self.scaler is not None):
        weights[self.scaler.state_dict_key] = self.scaler.state_dict()
    self.checkpoint.save(weights, outname)