FastSurferCNN.models.optimizer

FastSurferCNN.models.optimizer.get_optimizer(model, cfg)

Get an instance of requested optimizer.

Parameters:

modelFastSurferCNN, FastSurferVINN, torch.nn.DataParallel

The model for which an optimizer should be chosen.

cfgyacs.config.CfgNode

Configuration Node.

Returns:

torch.optim.optimizer.Optimizer

SGD, Adam, AdamW or rmsprop optimizer.

Raises:

NotImplementedError

Optimizer is not supported.