Commit 703dc8d7 authored by Konstantinos Bousmalis's avatar Konstantinos Bousmalis
Browse files

Fix provide_batch_fn for semi-supervised case.

parent 09bc6113
......@@ -180,7 +180,7 @@ def main(_):
if FLAGS.target_labeled_dataset != 'none':
# 1000 is the maximum number of labelled target samples that exists in
# the datasets.
target_semi_images, target_semi_labels = data_provider.provide(
target_semi_images, target_semi_labels = provide_batch_fn()(
FLAGS.target_labeled_dataset, 'train', FLAGS.batch_size)
# Calculate the proportion of source domain samples in the semi-
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment