GOOD.utils.train

Training utils.

Functions

at_stage(i, config)

Test if the current training stage at stage i.

nan2zero_get_mask(data, task, config)

Training data filter masks to process NAN.
GOOD.utils.train.at_stage(i, config)[source]

Test if the current training stage at stage i.

Parameters

  • i – Stage that is possibly 1, 2, 3, …

  • config – config object.

Returns: At stage i.

GOOD.utils.train.nan2zero_get_mask(data, task, config: Union[CommonArgs, Munch])[source]

Training data filter masks to process NAN.

Parameters

  • data (Batch) – input data

  • task (str) – mask function type

  • config (Union[CommonArgs, Munch]) – munchified dictionary of args (config.model.model_level)

Returns (Tensor):

[mask (Tensor) - NAN masks for data formats, targets (Tensor) - input labels]