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.