19 lines
440 B
Python
19 lines
440 B
Python
from federatedscope.register import register_trainer
|
|
from federatedscope.core.trainers import GeneralTorchTrainer
|
|
|
|
|
|
class CVTrainer(GeneralTorchTrainer):
|
|
"""
|
|
``CVTrainer`` is the same as ``core.trainers.GeneralTorchTrainer``.
|
|
"""
|
|
pass
|
|
|
|
|
|
def call_cv_trainer(trainer_type):
|
|
if trainer_type == 'cvtrainer':
|
|
trainer_builder = CVTrainer
|
|
return trainer_builder
|
|
|
|
|
|
register_trainer('cvtrainer', call_cv_trainer)
|