It looks to me like train_mixed_precision is not passed to train in model_builder.get_model so no matter what the config_sample says, there's no mixed precision from what I can tell. Doing mixed precision training seems to give a speedup of about 30%.
It looks to me like
train_mixed_precisionis not passed totraininmodel_builder.get_modelso no matter what theconfig_samplesays, there's no mixed precision from what I can tell. Doing mixed precision training seems to give a speedup of about 30%.