Skip to content

GitLab

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

Fix provide_batch_fn for semi-supervised case.

parent 09bc6113
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
domain_adaptation/domain_separation/dsn_train.py
View file @ 703dc8d7
...@@ -180,7 +180,7 @@ def main(_): ...@@ -180,7 +180,7 @@ def main(_):
if FLAGS.target_labeled_dataset != 'none': if FLAGS.target_labeled_dataset != 'none':
# 1000 is the maximum number of labelled target samples that exists in # 1000 is the maximum number of labelled target samples that exists in
# the datasets. # 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) FLAGS.target_labeled_dataset, 'train', FLAGS.batch_size)
# Calculate the proportion of source domain samples in the semi- # 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