Skip to content

Commit

Permalink
rc4 qat finetune data loading
Browse files Browse the repository at this point in the history
Summary:
With this diff, we are able to reproce the data loading based on the final rc4 model config

## Changes
1. Support category frequency as per instance weights instead of dataset level weights
2. Update rc4 qat finetune model config
3. Use naive sync BN in rc4 backbone

Reviewed By: stephenyan1231

Differential Revision: D56121350

fbshipit-source-id: 554446995ec9435ce439d7e454fe6ccf84a30934
  • Loading branch information
jiaxuzhu92 authored and facebook-github-bot committed Apr 17, 2024
1 parent 9ca6689 commit a59f056
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion detectron2/data/build.py
Original file line number Diff line number Diff line change
Expand Up @@ -484,7 +484,7 @@ def _train_loader_from_config(cfg, mapper=None, *, dataset=None, sampler=None):
repeat_factors = RepeatFactorTrainingSampler.repeat_factors_from_category_frequency(
dataset, cfg.DATALOADER.REPEAT_THRESHOLD, sqrt=cfg.DATALOADER.REPEAT_SQRT
)
sampler = RepeatFactorTrainingSampler(repeat_factors)
sampler = RepeatFactorTrainingSampler(repeat_factors, seed=cfg.SEED)
elif sampler_name == "RandomSubsetTrainingSampler":
sampler = RandomSubsetTrainingSampler(
len(dataset), cfg.DATALOADER.RANDOM_SUBSET_RATIO
Expand Down

0 comments on commit a59f056

Please sign in to comment.