17 lines
393 B
Python
17 lines
393 B
Python
from federatedscope.register import register_trainer
|
|
from federatedscope.core.trainers.torch_trainer import GeneralTorchTrainer
|
|
|
|
|
|
# Build your trainer here.
|
|
class MyTrainer(GeneralTorchTrainer):
|
|
pass
|
|
|
|
|
|
def call_my_trainer(trainer_type):
|
|
if trainer_type == 'mytrainer':
|
|
trainer_builder = MyTrainer
|
|
return trainer_builder
|
|
|
|
|
|
register_trainer('mytrainer', call_my_trainer)
|