mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix typro in hpo training

pull/365/head
liaoxingyu 2020-10-10 18:55:32 +08:00
parent 8d55a0bbd4
commit e5c3ff3b5b
2 changed files with 5 additions and 4 deletions
projects/HPOReID

2
projects/HPOReID/configs/baseline.yml
View File

@ -90,4 +90,4 @@ DATASETS:
CUDNN_BENCHMARK: True
OUTPUT_DIR: "projects/HPOReID/logs/dukemtmc/r34-ibn_bohb_bsz_num_"
OUTPUT_DIR: "projects/HPOReID/logs/dukemtmc/r34-ibn_bohb_bsz_num-inst"

7
projects/HPOReID/train_hpo.py
View File

@ -29,7 +29,7 @@ from fastreid.engine import DefaultTrainer, default_argument_parser, default_set
from fastreid.utils.events import CommonMetricPrinter
from fastreid.utils.file_io import PathManager
from naic.tune_hooks import TuneReportHook
from hporeid import *
logger = logging.getLogger("fastreid.project.tune")
@ -180,7 +180,8 @@ def main(args):
# CS.UniformIntegerHyperparameter(name="delay_iters", lower=20, upper=60),
# CS.UniformFloatHyperparameter(name="wd", lower=0, upper=1e-3),
# CS.UniformFloatHyperparameter(name="wd_bias", lower=0, upper=1e-3),
CS.CategoricalHyperparameter(name="bsz", choices=[64, 96, 128, 160, 224, 256]),
# CS.CategoricalHyperparameter(name="bsz", choices=[64, 96, 128, 160, 224, 256]),
CS.CategoricalHyperparameter(name="bsz", choices=[32, 64]),
CS.CategoricalHyperparameter(name="num_inst", choices=[2, 4, 8, 16, 32]),
# CS.UniformFloatHyperparameter(name="ce_scale", lower=0.1, upper=1.0),
# CS.UniformIntegerHyperparameter(name="circle_scale", lower=8, upper=256),
@ -212,7 +213,7 @@ def main(args):
cfg=cfg),
resources_per_trial={"cpu": 12, "gpu": 1},
search_alg=search_algo,
num_samples=args.num_samples,
num_samples=args.num_trials,
scheduler=scheduler,
progress_reporter=reporter,
local_dir=cfg.OUTPUT_DIR,