gigl.src.training.v1.lib.setup_model_device ============================================================ .. currentmodule:: gigl.src.training.v1.lib.training_process .. autofunction:: setup_model_device