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)