mmagic.utils.sampler

Module Contents

Classes

ArgumentsSampler	Dummy sampler only return input args multiple times.
NoiseSampler	Noise sampler to by call models.noise_fn to generate noise.
DataSampler	Sampler loop the train_dataloader.
ValDataSampler	Sampler loop the val_dataloader.

Functions

_check_keys(→ None)	Check whether target key is in the sample_kwargs.
get_sampler(sample_kwargs, runner)	Get a sampler to loop input data.

mmagic.utils.sampler._check_keys(sample_kwargs: dict, key: str) → None

Check whether target key is in the sample_kwargs.

mmagic.utils.sampler.get_sampler(sample_kwargs: dict, runner: Optional[mmengine.runner.Runner])

Get a sampler to loop input data.

Parameters

• sample_kwargs (dict) – _description_

• runner (Optional[Runner]) – _description_

Returns

_description_

Return type

_type_

class mmagic.utils.sampler.ArgumentsSampler(sample_kwargs: dict, runner: Optional[mmengine.runner.Runner] = None)

Dummy sampler only return input args multiple times.

__iter__()

__next__()

class mmagic.utils.sampler.NoiseSampler(sample_kwargs: dict, runner: mmengine.runner.Runner)

Noise sampler to by call models.noise_fn to generate noise.

__iter__()

__next__()

class mmagic.utils.sampler.DataSampler(sample_kwargs: dict, runner: mmengine.runner.Runner)

Sampler loop the train_dataloader.

__iter__()

__next__()

class mmagic.utils.sampler.ValDataSampler(sample_kwargs: dict, runner: mmengine.runner.Runner)

Sampler loop the val_dataloader.

__iter__()

__next__()