retraining original abl_all_post

2023-01-25 22:30:46 +00:00
parent ffdc60062d
commit 0bc5633b6c
3 changed files with 5 additions and 5 deletions
@@ -102,7 +102,7 @@ if __name__ == '__main__':
    ids = [id.split('_')[0] for id in ids]
    ids = set(ids)

-    iteration = ''
+    iteration = '70000'

    load_model(model, 'checkpoints/post/' + checkpoint_name, iteration=iteration)
    # load_model(model, 'checkpoints/' + 'abl_all_120k')