2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed function call

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-02-08 18:12:33 +01:00
parent d9a35893af
commit ea1b4943cf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/aae/train.py
View File

@ -564,7 +564,7 @@ if __name__ == "__main__":
tf.enable_eager_execution()
inlier_classes = [0]
iteration = 1
train_dataset, _ = prepare_training_data(test_fold_id=0, inlier_classes=[0],
train_dataset, _ = prepare_training_data(test_fold_id=0, inlier_classes=inlier_classes,
total_classes=10)
train_summary_writer = summary_ops_v2.create_file_writer(
'./summaries/train/number-' + str(inlier_classes[0]) + '/' + str(iteration))