30a0e7b4b572b1a48d64d2f7f3574493fd3c7d56,train.py,,train,#Any#,32

Before Change

    iteration = infos["iter"]
    epoch = infos["epoch"]
    loader.iterators = infos.get("iterators", loader.iterators)
    loader.split_ix = infos.get("split_ix", loader.split_ix)
    if opt.load_best_score == 1:
        best_val_score = infos.get("best_val_score", None)
    if opt.noamopt:
        optimizer._step = iteration

After Change

    // For back compatibility
    if "iterators" in infos:
        infos["loader_state_dict"] = {split: {"index_list": infos["split_ix"][split], "iter_counter": infos["iterators"][split]} for split in ["train", "val", "test"]}
    loader.load_state_dict(infos["loader_state_dict"])
    if opt.load_best_score == 1:
        best_val_score = infos.get("best_val_score", None)
    if opt.noamopt:
        optimizer._step = iteration

Instances