From 03f720616d76aa2262587ab73558be30fc23f01d Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sun, 15 Jan 2023 15:56:04 -0500 Subject: [PATCH] update training data --- cli/losschart.py | 1 + cli/train.json | 6 ++++-- config.json | 3 ++- 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/cli/losschart.py b/cli/losschart.py index 4ad919b4f..a96271f6f 100755 --- a/cli/losschart.py +++ b/cli/losschart.py @@ -82,6 +82,7 @@ def plot(logdir: str, name: str): # learning rate ax2 = ax1.twinx() ax2.set_ylabel('learning rate'.upper(), color='cyan') + ax2.set_ylim(0, 0.011) ax2.plot(step, rate, color='cyan', linewidth=3, linestyle='dashed') ax2.tick_params(axis='y', labelcolor='cyan') diff --git a/cli/train.json b/cli/train.json index 4481066ee..e6dad5e44 100644 --- a/cli/train.json +++ b/cli/train.json @@ -13,6 +13,7 @@ "init_text": "*" }, "preprocess": { + "id_task": 0, "process_src": "", "process_dst": "", "process_width": 512, @@ -31,10 +32,11 @@ "overlap_ratio": 0.2 }, "train_embedding": { + "id_task": 0, "embedding_name": "", - "learn_rate": "0.010:10, 0.008:20, 0.006:40, 0.004:80, 0.002:120, 0.001:160, 0.0005:200", + "learn_rate": "0.010:10, 0.008:50, 0.006:100, 0.004:200, 0.002:300, 0.001:400, 0.0005:500", "batch_size": 2, - "steps": 200, + "steps": 500, "data_root": "", "log_directory": "train/log", "template_filename": "subject_filewords.txt", diff --git a/config.json b/config.json index 539c07b86..8e31445b9 100644 --- a/config.json +++ b/config.json @@ -158,5 +158,6 @@ "images_history_pages_perload": 20.0, "inspiration_max_samples": 4, "inspiration_rows_num": 4, - "inspiration_cols_num": 6 + "inspiration_cols_num": 6, + "live_preview_refresh_period": 1000 } \ No newline at end of file