ggt.utils.model_utils
Module Contents
Functions
|
Get output shape of a PyTorch model or layer |
|
Enable random dropout during inference. From StackOverflow #63397197 |
|
Specify the dropout rate of all layers |
- ggt.utils.model_utils.get_output_shape(model, image_dim)
Get output shape of a PyTorch model or layer
- ggt.utils.model_utils.enable_dropout(model)
Enable random dropout during inference. From StackOverflow #63397197
- ggt.utils.model_utils.specify_dropout_rate(model, rate)
Specify the dropout rate of all layers