experiments/codes/experiment/checkpointable_multitask_experiment.py [58:79]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if skip_composition_registry:
            optimizer = Net.register_params_to_optimizer(
                self.config,
                self.representation_fn.model.weights,
                is_signature=self.is_signature,
            )
        elif skip_representation_registry:
            optimizer = Net.register_params_to_optimizer(
                self.config,
                self.composition_fn.model.weights,
                is_signature=self.is_signature,
            )
        else:
            optimizer = Net.register_params_to_optimizer(
                self.config,
                self.composition_fn.model.weights
                + self.representation_fn.model.weights,
                is_signature=self.is_signature,
            )
        self.optimizers, self.schedulers = Net.register_optimizers_to_schedulers(
            self.config, [optimizer]
        )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



experiments/codes/experiment/inference.py [61:82]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if skip_composition_registry:
            optimizer = Net.register_params_to_optimizer(
                self.config,
                self.representation_fn.model.weights,
                is_signature=self.is_signature,
            )
        elif skip_representation_registry:
            optimizer = Net.register_params_to_optimizer(
                self.config,
                self.composition_fn.model.weights,
                is_signature=self.is_signature,
            )
        else:
            optimizer = Net.register_params_to_optimizer(
                self.config,
                self.composition_fn.model.weights
                + self.representation_fn.model.weights,
                is_signature=self.is_signature,
            )
        self.optimizers, self.schedulers = Net.register_optimizers_to_schedulers(
            self.config, [optimizer]
        )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -