Function MSModelSetupVirtualBatch

• Defined in File model_c.h

Function Documentation

MSStatus MSModelSetupVirtualBatch(MSModelHandle model, int virtual_batch_multiplier, float lr, float momentum)

Setup training with virtual batches. Only valid for Lite Train.

Parameters

• model – [in] Model object handle.

• virtual_batch_multiplier – [in] - virtual batch multiplier, use any number < 1 to disable.

• lr – [in] - learning rate to use for virtual batch, -1 for internal configuration.

• momentum – [in] - batch norm momentum to use for virtual batch, -1 for internal configuration.

Returns

MSStatus.