13 lines
408 B
Python
13 lines
408 B
Python
from federatedscope.register import register_trainer
|
|
from federatedscope.core.trainers import BaseTrainer
|
|
|
|
|
|
def call_local_entropy_trainer(trainer_type):
|
|
if trainer_type == 'local_entropy_trainer':
|
|
from federatedscope.contrib.trainer.local_entropy \
|
|
import LocalEntropyTrainer
|
|
return LocalEntropyTrainer
|
|
|
|
|
|
register_trainer('local_entropy_trainer', call_local_entropy_trainer)
|