azula.plugins.utils¶

Miscelaneous plugin helpers.

Functions¶

load_cards

Returns the name-card mapping of pre-trained models available in a plugin.

Descriptions¶

azula.plugins.utils.load_cards(plugin)¶[source]

Returns the name-card mapping of pre-trained models available in a plugin.

Parameters:: plugin (ModuleType) – The plugin module.

Example

>>> cards = load_cards(azula.plugins.adm)
>>> list(cards.keys())
['imagenet_64x64_cond',
 'imagenet_128x128_cond',
 'imagenet_256x256',
 'imagenet_256x256_cond',
 'imagenet_512x512_cond',
 'ffhq_256x256']