INNER CODE UNIT · Python
set_train_type
tensorlayer/HyperPose · hyperpose/Config/__init__.py:304
def set_train_type(train_type):
'''set single_train or parallel train
default using single train, which train the model on one GPU.
set parallel train will use Kungfu library to accelerate training on multiple GPU.
to use parallel train better, it is also allow to set parallel training optimizor by set_kungfu_option.
Parameters
----------
arg1 : Config.TRAIN
a enum value of enum class Config.TRAIN,available options:
| Config.TRAIN.Single_train
| Config.TRAIN.Parallel_train
Returns
-------
None