Skip to content
This repository has been archived by the owner on Apr 17, 2023. It is now read-only.

Feature/val batch and seed #69

Merged
merged 3 commits into from
Oct 18, 2022
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions mpa/det/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,6 +69,7 @@ def run(self, model_cfg, model_ckpt, data_cfg, **kwargs):

# Data
datasets = [build_dataset(cfg.data.train)]
cfg.data.val.samples_per_gpu = cfg.data.samples_per_gpu
harimkang marked this conversation as resolved.
Show resolved Hide resolved

if hasattr(cfg, 'hparams'):
if cfg.hparams.get('adaptive_anchor', False):
Expand Down
1 change: 1 addition & 0 deletions recipes/stages/classification/train.yaml
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
_base_: [
'../_base_/default.py',
'../_base_/logs/tensorboard_logger.py',
'../_base_/optimizers/sgd.py',
'../_base_/runners/epoch_runner_cancel.py',
Expand Down