Set default models to training mode in the train_step.
#90
| Job | Run time |
|---|---|
| 2m 32s | |
| 2m 32s |
training mode in the train_step.
#90
| Job | Run time |
|---|---|
| 2m 32s | |
| 2m 32s |