FS-TFP/federatedscope/gfl/trainer/__init__.py

15 lines
527 B
Python

from __future__ import absolute_import
from __future__ import print_function
from __future__ import division
from federatedscope.gfl.trainer.graphtrainer import GraphMiniBatchTrainer
from federatedscope.gfl.trainer.linktrainer import LinkFullBatchTrainer, \
LinkMiniBatchTrainer
from federatedscope.gfl.trainer.nodetrainer import NodeFullBatchTrainer, \
NodeMiniBatchTrainer
__all__ = [
'GraphMiniBatchTrainer', 'LinkFullBatchTrainer', 'LinkMiniBatchTrainer',
'NodeFullBatchTrainer', 'NodeMiniBatchTrainer'
]