Misc

biapy.utils.misc.setup_for_distributed(is_master)[source]

This function disables printing when not in master process

biapy.utils.misc.is_dist_avail_and_initialized()[source]
biapy.utils.misc.get_world_size()[source]
biapy.utils.misc.get_rank()[source]
biapy.utils.misc.is_main_process()[source]
biapy.utils.misc.init_devices(args, cfg)[source]
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.get_grad_norm_(parameters, norm_type: float = 2.0) Tensor[source]
biapy.utils.misc.save_model(cfg, jobname, epoch, model, model_without_ddp, optimizer, loss_scaler)[source]
biapy.utils.misc.save_on_master(*args, **kwargs)[source]
biapy.utils.misc.get_checkpoint_path(cfg, jobname)[source]
biapy.utils.misc.load_model_checkpoint(cfg, jobname, model_without_ddp, device, optimizer=None, loss_scaler=None)[source]
biapy.utils.misc.all_reduce_mean(x)[source]
biapy.utils.misc.to_pytorch_format(x, axis_order, device, dtype=torch.float32)[source]
biapy.utils.misc.to_numpy_format(x, axis_order_back)[source]
biapy.utils.misc.time_text(t)[source]
class biapy.utils.misc.NativeScalerWithGradNormCount[source]

Bases: object

state_dict_key = 'amp_scaler'
state_dict()[source]
load_state_dict(state_dict)[source]
class biapy.utils.misc.TensorboardLogger(log_dir)[source]

Bases: object

set_step(step=None)[source]
update(head='scalar', step=None, **kwargs)[source]
flush()[source]
class biapy.utils.misc.SmoothedValue(window_size=20, fmt=None)[source]

Bases: object

Track a series of values and provide access to smoothed values over a window or the global series average.

update(value, n=1)[source]
synchronize_between_processes()[source]

Warning: does not synchronize the deque!

property median
property avg
property global_avg
property max
property value
class biapy.utils.misc.MetricLogger(delimiter='\t', verbose=False)[source]

Bases: object

update(**kwargs)[source]
synchronize_between_processes()[source]
add_meter(name, meter)[source]
log_every(iterable, print_freq, header=None)[source]