Cool_Tools/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of synced 2025-02-22 05:09:58 +01:00
Code Releases Wiki Activity

Update utils.py

Browse Source
This commit is contained in:
liujing04 2023-04-09 15:52:48 +00:00 committed by GitHub
parent 11c871d63e
commit dce1baefb3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
train/utils.py
View File

@ -50,7 +50,7 @@ def load_checkpoint_d(checkpoint_path, combd,sbd, optimizer=None,load_opt=1):
optimizer.load_state_dict(checkpoint_dict['optimizer']) optimizer.load_state_dict(checkpoint_dict['optimizer'])
# except: # except:
# traceback.print_exc() # traceback.print_exc()
logger.info("Loaded checkpoint '{}' (iteration {})" .format(checkpoint_path, iteration)) logger.info("Loaded checkpoint '{}' (epoch {})" .format(checkpoint_path, iteration))
return model, optimizer, learning_rate, iteration return model, optimizer, learning_rate, iteration
@ -80,7 +80,7 @@ def load_checkpoint_d(checkpoint_path, combd,sbd, optimizer=None,load_opt=1):
# model.module.load_state_dict(new_state_dict) # model.module.load_state_dict(new_state_dict)
# else: # else:
# model.load_state_dict(new_state_dict) # model.load_state_dict(new_state_dict)
# logger.info("Loaded checkpoint '{}' (iteration {})" .format( # logger.info("Loaded checkpoint '{}' (epoch {})" .format(
# checkpoint_path, iteration)) # checkpoint_path, iteration))
# return model, optimizer, learning_rate, iteration # return model, optimizer, learning_rate, iteration
def load_checkpoint(checkpoint_path, model, optimizer=None,load_opt=1): def load_checkpoint(checkpoint_path, model, optimizer=None,load_opt=1):
@ -116,7 +116,7 @@ def load_checkpoint(checkpoint_path, model, optimizer=None,load_opt=1):
optimizer.load_state_dict(checkpoint_dict['optimizer']) optimizer.load_state_dict(checkpoint_dict['optimizer'])
# except: # except:
# traceback.print_exc() # traceback.print_exc()
logger.info("Loaded checkpoint '{}' (iteration {})" .format(checkpoint_path, iteration)) logger.info("Loaded checkpoint '{}' (epoch {})" .format(checkpoint_path, iteration))
return model, optimizer, learning_rate, iteration return model, optimizer, learning_rate, iteration