Misc
- biapy.utils.misc.setup_for_distributed(is_master)[source]
This function disables printing when not in master process
- biapy.utils.misc.set_seed(seed=42)[source]
Sets the seed on multiple python modules to obtain results as reproducible as possible.
- Parameters:
seed (int, optional) – Seed value.
- biapy.utils.misc.save_model(cfg, jobname, epoch, model, model_without_ddp, optimizer, loss_scaler)[source]
- biapy.utils.misc.load_model_checkpoint(cfg, jobname, model_without_ddp, device, optimizer=None, loss_scaler=None)[source]
- class biapy.utils.misc.NativeScalerWithGradNormCount[source]
Bases:
object
- state_dict_key = 'amp_scaler'