Init

biapy.models.build_model(cfg, job_identifier, device)[source]

Build selected model

Parameters:

  • cfg (YACS CN object) – Configuration.

  • job_identifier (str) – Job name.

  • device (Torch device) – Using device (“cpu” or “cuda” for GPU).

Returns:

model – Selected model.

Return type:

Keras model

biapy.models.build_torchvision_model(cfg, device)[source]