Skip to content

v1.0.1

Latest
Compare
Choose a tag to compare
@djanibekov djanibekov released this 04 Feb 10:39
· 11 commits to main since this release

Resolve previous release bug with internal checkpoint's namings

checkpoint['cfg']['checkpoint']['finetune_from_model'] = None
checkpoint['cfg']['checkpoint']['save_dir'] = None
checkpoint['cfg']['common']['tensorboard_logdir'] = None
checkpoint['cfg']['common']['user_dir'] = None
checkpoint['cfg']['criterion']['_name'] = "artst"

checkpoint['cfg']['task']._name = "artst"
checkpoint['cfg']['task'].arch = 'artst_transformer_base_asr'
checkpoint['cfg']['task'].bpe_tokenizer = None
checkpoint['cfg']['task'].criterion = "artst"
checkpoint['cfg']['task'].data = None
checkpoint['cfg']['task'].finetuned_from_model = None
checkpoint['cfg']['task'].hubert_label_dir = None
checkpoint['cfg']['task'].save_dir = None
checkpoint['cfg']['task'].task = "artst"
checkpoint['cfg']['task'].tensorboard_logdir = None
checkpoint['cfg']['task'].user_dir = None


checkpoint['cfg']['model']._name = "artst_transformer_base_asr"
checkpoint['cfg']['model'].arch = "artst_transformer_base_asr"
checkpoint['cfg']['model'].bpe_tokenizer = None
checkpoint['cfg']['model'].criterion = "artst"
checkpoint['cfg']['model'].data = None
checkpoint['cfg']['model'].finetuned_from_model = None
checkpoint['cfg']['model'].hubert_label_dir = None
checkpoint['cfg']['model'].save_dir = None
checkpoint['cfg']['model'].task = "artst"
checkpoint['cfg']['model'].tensorboard_logdir = None
checkpoint['cfg']['model'].user_dir = None

checkpoint['optimizer_history'] = [{**s, "criterion_name": "ArTSTCriterion"} for s in checkpoint['optimizer_history']]