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

Improved saving location of summaries

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-02-08 11:37:35 +01:00
parent 5ba3bc552f
commit fc1aa20396
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -399,6 +399,9 @@ def extract_batch(data: np.ndarray, it: int, batch_size: int) -> tfe.Variable:
if __name__ == "__main__":
tf.enable_eager_execution()
train_summary_writer = summary_ops_v2.create_file_writer('./summaries/train')
inlier_classes = [0]
iteration = 1
train_summary_writer = summary_ops_v2.create_file_writer(
'./summaries/train/number-' + str(inlier_classes[0]) + '/' + str(iteration))
with train_summary_writer.as_default(), summary_ops_v2.always_record_summaries():
train_mnist(folding_id=0, inlier_classes=[0], total_classes=10)
train_mnist(folding_id=0, inlier_classes=inlier_classes, total_classes=10)