a4217c8986523a59dd9b764467fc21b29dd572ba,train.py,,run,#Any#Any#Any#Any#Any#Any#,44

Before Change

    // Check if training dir exists
    assert os.path.isdir(train_images_dir)
    if eval_dir is not None:
        eval_images_dir, eval_labels_dir = os.path.join(eval_dir, "images"), os.path.join(eval_dir, "labels")
        assert os.path.isdir(eval_images_dir)

    for i in trange(0, training_params.n_epochs, training_params.evaluate_every_epoch, desc="Evaluated epochs"):
        // Train for one epoch

After Change

            assert os.path.isdir(eval_images_dir)
            filenames_evaluation = glob(os.path.join(eval_images_dir, "*.jpg")) \
                                   + glob(os.path.join(eval_images_dir, "*.png"))
    elif os.path.isfile(train_dir_or_csv):
        train_input = train_dir_or_csv
        train_labels_input = None
        eval_input = eval_dir_or_csv

Instances