Created by: ccmeteorljh

在train.py中 dataset_dir 路径写死了

def prepare_reader(epoch_id, train_py_reader, train_bs, val_bs, trn_dir,
                   img_dim, min_scale, rect_val):
    train_reader = reader.train(
        traindir="/data/imagenet/%strain" % trn_dir,
        sz=img_dim,
        min_scale=min_scale,
        shuffle_seed=epoch_id + 1)
    train_py_reader.decorate_paddle_reader(
        paddle.batch(
            train_reader, batch_size=train_bs))

    test_reader = reader.test(
        valdir="/data/imagenet/%svalidation" % trn_dir,
        bs=val_bs * DEVICE_NUM,
        sz=img_dim,
        rect_val=rect_val)