avalanche.models.packnet.packnet_simple_mlp

avalanche.models.packnet.packnet_simple_mlp(num_classes=10, input_size=784, hidden_size=512, hidden_layers=1, drop_rate=0.5) → PackNetModel[source]

Convenience function for creating a PackNet compatible SimpleMLP model.

Parameters:

  • num_classes – output size

  • input_size – input size

  • hidden_size – hidden layer size

  • hidden_layers – number of hidden layers

  • drop_rate – dropout rate. 0 to disable

Returns:

A PackNet compatible model