12 lines
326 B
Python
12 lines
326 B
Python
from federatedscope.register import register_trainer
|
|
from federatedscope.core.trainers import BaseTrainer
|
|
|
|
|
|
def call_sam_trainer(trainer_type):
|
|
if trainer_type == 'sam_trainer':
|
|
from federatedscope.contrib.trainer.sam import SAMTrainer
|
|
return SAMTrainer
|
|
|
|
|
|
register_trainer('sam_trainer', call_sam_trainer)
|