From a4167b59d65d6d669fd1e902467a59385ce03f9b Mon Sep 17 00:00:00 2001 From: unanmed <1319491857@qq.com> Date: Sat, 22 Mar 2025 18:50:35 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E4=B8=80=E8=BD=AE=E5=AF=B9=E6=8A=97?= =?UTF-8?q?=E8=AE=AD=E7=BB=83=E7=9A=84=20sh=20=E8=84=9A=E6=9C=AC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cycle.sh | 15 +++++++++++++++ ginka/validate.py | 2 +- 2 files changed, 16 insertions(+), 1 deletion(-) create mode 100644 cycle.sh diff --git a/cycle.sh b/cycle.sh new file mode 100644 index 0000000..b4e5c22 --- /dev/null +++ b/cycle.sh @@ -0,0 +1,15 @@ +# 训练部分 +python3 -m minamo.train --epochs 30 --resume true +python3 -m minamo.train --epochs 10 --resume true --train "datasets/minamo-dataset-1.json" --validate "datasets/minamo-eval-1.json" +python3 -m minamo.train --epochs 10 --resume true +python3 -m ginka.train --epochs 10 --resume true +python3 -m ginka.validate +# 训练完毕,处理数据 +mv "minamo-dataset.json" "datasets/minamo-dataset-$1.json" +mv "minamo-eval.json" "datasets/minamo-eval-$1.json" +cd data +pnpm minamo "../minamo-dataset.json" "../result/ginka_val.json" "../../Apeiria/project" assigned +pnpm minamo "../minamo-dataset.json" "../result/ginka_val.json" "../../Apeiria-eval/project" assigned:100:10 +pnpm merge "../datasets/minamo-dataset-merged.json" "../datasets/minamo-dataset-merged.json" "../datasets/minamo-dataset-$1.json" +pnpm merge "../datasets/minamo-eval-merged.json" "../datasets/minamo-eval-merged.json" "../datasets/minamo-eval-$1.json" +cd .. diff --git a/ginka/validate.py b/ginka/validate.py index e100603..0eca862 100644 --- a/ginka/validate.py +++ b/ginka/validate.py @@ -66,7 +66,7 @@ def matrix_to_image_cv(map_matrix, tile_set, tile_size=32): def validate(): print(f"Using {'cuda' if torch.cuda.is_available() else 'cpu'} to validate model.") model = GinkaModel() - state = torch.load("result/ginka.pth", map_location=device)["model_state"] + state = torch.load("result/ginka_checkpoint/10.pth", map_location=device)["model_state"] model.load_state_dict(state) model.to(device)