Full fix
This commit is contained in:
9
train.py
9
train.py
@@ -292,6 +292,8 @@ def main(args):
|
||||
num_workers=args.num_workers,
|
||||
image_size=args.image_size,
|
||||
build_test=False, # оценка идёт через multi-positive eval ниже
|
||||
mutually_exclusive=args.mutually_exclusive,
|
||||
seed=args.seed,
|
||||
)
|
||||
|
||||
# Multi-positive eval (протокол §6.2): уникальная галерея + positive-карты.
|
||||
@@ -485,7 +487,7 @@ def parse_args():
|
||||
p.add_argument("--text_levels", nargs="+", default=["level1"])
|
||||
p.add_argument("--train_meta", default="cross-area-drone2sate-train.json")
|
||||
p.add_argument("--test_meta", default="cross-area-drone2sate-test.json")
|
||||
p.add_argument("--image_size", type=int, default=384)
|
||||
p.add_argument("--image_size", type=int, default=256)
|
||||
p.add_argument("--num_workers", type=int, default=8)
|
||||
|
||||
# Model
|
||||
@@ -496,6 +498,11 @@ def parse_args():
|
||||
p.add_argument("--shared_dim", type=int, default=512)
|
||||
p.add_argument("--inject_mona", action="store_true", default=True)
|
||||
p.add_argument("--mona_bottleneck", type=int, default=64)
|
||||
p.add_argument(
|
||||
"--mutually_exclusive", action=argparse.BooleanOptionalAction, default=True,
|
||||
help="Батчи без общих позитивных тайлов (§6.4); --no-mutually_exclusive "
|
||||
"→ обычный random shuffle.",
|
||||
)
|
||||
|
||||
# Training
|
||||
p.add_argument("--epochs", type=int, default=50)
|
||||
|
||||
Reference in New Issue
Block a user