deepdisc.model.models

Functions

return_lazy_model(cfg[, freeze])

Return a model formed from a LazyConfig with the backbone

Module Contents

return_lazy_model(cfg, freeze=True)[source]

Return a model formed from a LazyConfig with the backbone frozen. Only the head layers will be trained.

Parameters:

cfg (.py file) – a LazyConfig

Return type:

torch model