Esempio n. 1
0
def test_set_obj_train_config():
    fed_client = FederatedClient()
    fed_client.optimizer = True

    train_config = TrainConfig(id=100, model=None, loss_fn=None)

    fed_client.set_obj(train_config)

    assert fed_client.train_config.id == train_config.id
    assert fed_client.optimizer is None
Esempio n. 2
0
 def set_train_config(self, *args, **kwargs):
     config = TrainConfig(*args, **kwargs)
     self._train_config_ptr = config.send(self)