ggt.utils.model_utils

Module Contents

Functions

get_output_shape(model, image_dim)

Get output shape of a PyTorch model or layer

enable_dropout(model)

Enable random dropout during inference. From StackOverflow #63397197

specify_dropout_rate(model, rate)

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