def _construct_examples_batch(batch_size, split, num_classes, num_tr_examples_per_class, num_val_examples_per_class): data_provider = data.DataProvider(split, config.get_data_config()) examples_batch = data_provider.get_batch(batch_size, num_classes, num_tr_examples_per_class, num_val_examples_per_class) return utils.unpack_data(examples_batch)
def _construct_examples_batch(batch_size, split, num_classes, num_tr_examples_per_class, num_val_examples_per_class, use_cross=False): data_provider = data.DataProvider(split, config.get_data_config(), feat_dim=FLAGS.feat_dim, use_cross=use_cross) examples_batch = data_provider.get_batch(batch_size, num_classes, num_tr_examples_per_class, num_val_examples_per_class, num_pretrain_classes=FLAGS.num_pretrain_classes) return utils.unpack_data(examples_batch)