MCPcopy Create free account

hub / github.com/AvivNavon/nash-mtl / functions

Functions112 in github.com/AvivNavon/nash-mtl

↓ 21 callersMethodconv_layer
(self, channel, pred=False)
experiments/nyuv2/models.py:131
↓ 12 callersMethodconv_layer
(self, channel)
experiments/nyuv2/models.py:397
↓ 12 callersMethodparameters
(self)
methods/weight_methods.py:439
↓ 11 callersMethod__init__
(self, n_tasks, device: torch.device, main_task)
methods/weight_methods.py:405
↓ 6 callersMethodatt_layer
(self, channel)
experiments/nyuv2/models.py:160
↓ 6 callersFunctioncalc_loss
(x_pred, x_output, task_type)
experiments/nyuv2/trainer.py:27
↓ 5 callersMethodbackward
( self, losses: torch.Tensor, parameters: Union[List[torch.nn.parameter.Parameter], to
methods/weight_methods.py:521
↓ 4 callersMethodbackward
( self, losses, **kwargs )
methods/weight_methods.py:805
↓ 3 callersMethod__init__
(self)
experiments/nyuv2/models.py:470
↓ 3 callersMethod_min_norm_element_from2
Analytical solution for min_{c} |cx_1 + (1-c)x_2|_2^2 d is the distance (objective) optimzed v1v1 = <x1,x1> v1v2 = <x
methods/min_norm_solvers.py:15
↓ 3 callersFunctionextract_weight_method_parameters_from_args
(args)
experiments/utils.py:115
↓ 3 callersMethodparameters
(self)
methods/weight_methods.py:813
↓ 3 callersFunctionset_logger
()
experiments/utils.py:84
↓ 3 callersMethodupdate
(self, pred, target)
experiments/nyuv2/utils.py:10
↓ 2 callersMethod_min_norm_2d
Find the minimum norm solution as combination of two points This is correct only in 2D ie. min_c |\sum c_i x_i|_2^2 st. \sum
methods/min_norm_solvers.py:39
↓ 2 callersMethodbatch_forward
(self, x)
experiments/toy/problem.py:41
↓ 2 callersFunctiondepth_error
(x_pred, x_output)
experiments/nyuv2/utils.py:26
↓ 2 callersFunctionevaluate
(model, loader, std, scale_target)
experiments/quantum_chemistry/trainer.py:34
↓ 2 callersFunctionget_device
(no_cuda=False, gpus="0")
experiments/utils.py:109
↓ 2 callersMethodget_metrics
(self)
experiments/nyuv2/utils.py:19
↓ 2 callersFunctionnormal_error
(x_pred, x_output)
experiments/nyuv2/utils.py:40
↓ 2 callersFunctionset_seed
for reproducibility :param seed: :return:
experiments/utils.py:91
↓ 1 callersMethod_calc_phi_alpha_linearization
(self)
methods/weight_methods.py:157
↓ 1 callersMethod_init_optim_problem
(self)
methods/weight_methods.py:163
↓ 1 callersMethod_init_task_heads
(self)
experiments/quantum_chemistry/models.py:26
↓ 1 callersMethod_next_point
(cur_val, grad, n)
methods/min_norm_solvers.py:93
↓ 1 callersMethod_project_conflicting
(self, grads: List[Tuple[torch.Tensor]])
methods/weight_methods.py:497
↓ 1 callersMethod_projection2simplex
Given y, it solves argmin_z |y-z|_2 st \sum z = 1 , 1 >= z_i >= 0 for all i
methods/min_norm_solvers.py:76
↓ 1 callersMethod_set_pc_grads
(self, losses, shared_parameters, task_specific_parameters=None)
methods/weight_methods.py:473
↓ 1 callersMethod_stop_criteria
(self, gtg, alpha_t)
methods/weight_methods.py:123
↓ 1 callersMethodbackward
Parameters ---------- losses : shared_parameters : task_specific_parameters : last_shared_parameters
methods/weight_methods.py:35
↓ 1 callersMethodcagrad
(self, grads, alpha=0.5, rescale=1)
methods/weight_methods.py:576
↓ 1 callersFunctiondelta_fn
(a)
experiments/nyuv2/utils.py:81
↓ 1 callersFunctiondelta_fn
(a)
experiments/quantum_chemistry/utils.py:91
↓ 1 callersMethodfind_min_norm_element
Given a list of vectors (vecs), this method finds the minimum norm element in the convex hull as min |u|_2 st. u = \sum c_i vecs[i] a
methods/min_norm_solvers.py:110
↓ 1 callersMethodget_weighted_loss
( self, losses: torch.Tensor, shared_parameters: Union[List[torch.nn.parameter.Paramet
methods/weight_methods.py:22
↓ 1 callersMethodget_weighted_loss
Parameters ---------- losses : shared_parameters : shared parameters kwargs : Returns -----
methods/weight_methods.py:190
↓ 1 callersMethodget_weighted_loss
(self, losses: torch.Tensor, **kwargs)
methods/weight_methods.py:411
↓ 1 callersMethodget_weighted_loss
Parameters ---------- losses : shared_parameters : shared parameters kwargs : Returns -------
methods/weight_methods.py:542
↓ 1 callersMethodgrad2vec
(shared_params, grads, grad_dims, task)
methods/weight_methods.py:612
↓ 1 callersFunctiongradient_normalizers
(grads, losses, normalization_type)
methods/min_norm_solvers.py:201
↓ 1 callersMethodlast_shared_parameters
Parameters of the last shared layer. Returns -------
experiments/nyuv2/models.py:480
↓ 1 callersMethodlast_shared_parameters
(self)
experiments/quantum_chemistry/models.py:63
↓ 1 callersFunctionmain
(path, lr, bs, device)
experiments/nyuv2/trainer.py:52
↓ 1 callersFunctionmain
( data_path: str, batch_size: int, device: torch.device, method: str, weight_method_params
experiments/quantum_chemistry/trainer.py:67
↓ 1 callersFunctionmain
(method_type, device, n_iter, scale)
experiments/toy/trainer.py:23
↓ 1 callersMethodoverwrite_grad
(self, shared_parameters, newgrad, grad_dims)
methods/weight_methods.py:628
↓ 1 callersFunctionplot_2d_pareto
Adaptation of code from: https://github.com/Cranial-XIX/CAGrad
experiments/toy/utils.py:11
↓ 1 callersFunctionrequirements
()
setup.py:14
↓ 1 callersMethodshared_modules
(self)
experiments/nyuv2/models.py:114
↓ 1 callersMethodshared_parameters
(self)
experiments/nyuv2/models.py:474
↓ 1 callersMethodshared_parameters
(self)
experiments/quantum_chemistry/models.py:51
↓ 1 callersMethodsolve_optimization
(self, gtg: np.array)
methods/weight_methods.py:133
↓ 1 callersMethodtask_specific_parameters
(self)
experiments/nyuv2/models.py:477
↓ 1 callersMethodtask_specific_parameters
(self)
experiments/quantum_chemistry/models.py:60
Method__call__
(self, img, label, depth, normal)
experiments/nyuv2/data.py:23
Method__call__
(self, data)
experiments/quantum_chemistry/utils.py:14
Method__call__
(self, data)
experiments/quantum_chemistry/utils.py:21
Method__call__
( self, losses: torch.Tensor, shared_parameters: Union[ List[torch.nn.para
methods/weight_methods.py:76
Method__ceil__
(self, losses, **kwargs)
methods/weight_methods.py:810
Method__getitem__
(self, index)
experiments/nyuv2/data.py:83
Method__init__
(self, num_classes)
experiments/nyuv2/utils.py:6
Method__init__
(self)
experiments/nyuv2/models.py:11
Method__init__
(self)
experiments/nyuv2/models.py:288
Method__init__
(self, model_type="standard")
experiments/nyuv2/models.py:314
Method__init__
(self, scale=[1.0, 1.2, 1.5])
experiments/nyuv2/data.py:20
Method__init__
(self, root, train=True, augmentation=False)
experiments/nyuv2/data.py:67
Method__init__
(self, target: list = None)
experiments/quantum_chemistry/utils.py:7
Method__init__
(self, n_tasks, num_features=11, dim=64)
experiments/quantum_chemistry/models.py:11
Method__init__
(self, scale=1.0, scale_both_losses=1.0)
experiments/toy/problem.py:8
Method__init__
(self, n_tasks: int, device: torch.device)
methods/weight_methods.py:16
Method__init__
( self, n_tasks: int, device: torch.device, max_norm: float = 1.0, upd
methods/weight_methods.py:100
Method__init__
( self, n_tasks: int, device: torch.device, task_weights: Union[List[float], t
methods/weight_methods.py:278
Method__init__
( self, n_tasks: int, device: torch.device, task_weights: Union[List[float], t
methods/weight_methods.py:300
Method__init__
( self, n_tasks, device: torch.device, params="shared", normalization="none" )
methods/weight_methods.py:327
Method__init__
(self, n_tasks, device: torch.device)
methods/weight_methods.py:423
Method__init__
(self, n_tasks: int, device: torch.device, reduction="sum")
methods/weight_methods.py:455
Method__init__
(self, n_tasks, device: torch.device, c=0.4)
methods/weight_methods.py:538
Method__init__
(self, n_tasks, device: torch.device)
methods/weight_methods.py:660
Method__init__
(self, n_tasks, device: torch.device)
methods/weight_methods.py:674
Method__init__
Parameters ---------- n_tasks : iteration_window : 'iteration' loss is averaged over the last 'iteration_window' los
methods/weight_methods.py:749
Method__init__
:param method:
methods/weight_methods.py:794
Method__len__
(self)
experiments/nyuv2/data.py:118
Method_flattening
(grad)
methods/weight_methods.py:338
Methodbackward
( self, losses: torch.Tensor, shared_parameters: Union[ List[torch.nn.para
methods/weight_methods.py:246
Methodbackward
( self, losses: torch.Tensor, parameters: Union[List[torch.nn.parameter.Parameter], to
methods/weight_methods.py:641
Functioncount_parameters
(model)
experiments/utils.py:80
Methodfind_min_norm_element_FW
Given a list of vectors (vecs), this method finds the minimum norm element in the convex hull as min |u|_2 st. u = \sum c_i vecs[i] a
methods/min_norm_solvers.py:157
Methodforward
(self, x)
experiments/nyuv2/models.py:181
Methodforward
(self, x, return_representation=False)
experiments/nyuv2/models.py:305
Methodforward
(self, x)
experiments/nyuv2/models.py:430
Methodforward
(self, x, return_representation=False)
experiments/nyuv2/models.py:487
Methodforward
(self, data, return_representation=False)
experiments/quantum_chemistry/models.py:33
Methodforward
(self, x, compute_grad=False)
experiments/toy/problem.py:14
Methodget_weighted_loss
(self, losses, **kwargs)
methods/weight_methods.py:292
Methodget_weighted_loss
(self, losses, **kwargs)
methods/weight_methods.py:314
Methodget_weighted_loss
Parameters ---------- losses : shared_parameters : last_shared_parameters : representation :
methods/weight_methods.py:349
Methodget_weighted_loss
(self, losses: torch.Tensor, **kwargs)
methods/weight_methods.py:427
Methodget_weighted_loss
( self, losses: torch.Tensor, shared_parameters: Union[ List[torch.nn.para
methods/weight_methods.py:460
Methodget_weighted_loss
(self, losses: torch.Tensor, **kwargs)
methods/weight_methods.py:663
next →1–100 of 112, ranked by callers