Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/AvivNavon/nash-mtl
/ functions
Functions
112 in github.com/AvivNavon/nash-mtl
⨍
Functions
112
◇
Types & classes
25
↓ 21 callers
Method
conv_layer
(self, channel, pred=False)
experiments/nyuv2/models.py:131
↓ 12 callers
Method
conv_layer
(self, channel)
experiments/nyuv2/models.py:397
↓ 12 callers
Method
parameters
(self)
methods/weight_methods.py:439
↓ 11 callers
Method
__init__
(self, n_tasks, device: torch.device, main_task)
methods/weight_methods.py:405
↓ 6 callers
Method
att_layer
(self, channel)
experiments/nyuv2/models.py:160
↓ 6 callers
Function
calc_loss
(x_pred, x_output, task_type)
experiments/nyuv2/trainer.py:27
↓ 5 callers
Method
backward
( self, losses: torch.Tensor, parameters: Union[List[torch.nn.parameter.Parameter], to
methods/weight_methods.py:521
↓ 4 callers
Method
backward
( self, losses, **kwargs )
methods/weight_methods.py:805
↓ 3 callers
Method
__init__
(self)
experiments/nyuv2/models.py:470
↓ 3 callers
Method
_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 callers
Function
extract_weight_method_parameters_from_args
(args)
experiments/utils.py:115
↓ 3 callers
Method
parameters
(self)
methods/weight_methods.py:813
↓ 3 callers
Function
set_logger
()
experiments/utils.py:84
↓ 3 callers
Method
update
(self, pred, target)
experiments/nyuv2/utils.py:10
↓ 2 callers
Method
_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 callers
Method
batch_forward
(self, x)
experiments/toy/problem.py:41
↓ 2 callers
Function
depth_error
(x_pred, x_output)
experiments/nyuv2/utils.py:26
↓ 2 callers
Function
evaluate
(model, loader, std, scale_target)
experiments/quantum_chemistry/trainer.py:34
↓ 2 callers
Function
get_device
(no_cuda=False, gpus="0")
experiments/utils.py:109
↓ 2 callers
Method
get_metrics
(self)
experiments/nyuv2/utils.py:19
↓ 2 callers
Function
normal_error
(x_pred, x_output)
experiments/nyuv2/utils.py:40
↓ 2 callers
Function
set_seed
for reproducibility :param seed: :return:
experiments/utils.py:91
↓ 1 callers
Method
_calc_phi_alpha_linearization
(self)
methods/weight_methods.py:157
↓ 1 callers
Method
_init_optim_problem
(self)
methods/weight_methods.py:163
↓ 1 callers
Method
_init_task_heads
(self)
experiments/quantum_chemistry/models.py:26
↓ 1 callers
Method
_next_point
(cur_val, grad, n)
methods/min_norm_solvers.py:93
↓ 1 callers
Method
_project_conflicting
(self, grads: List[Tuple[torch.Tensor]])
methods/weight_methods.py:497
↓ 1 callers
Method
_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 callers
Method
_set_pc_grads
(self, losses, shared_parameters, task_specific_parameters=None)
methods/weight_methods.py:473
↓ 1 callers
Method
_stop_criteria
(self, gtg, alpha_t)
methods/weight_methods.py:123
↓ 1 callers
Method
backward
Parameters ---------- losses : shared_parameters : task_specific_parameters : last_shared_parameters
methods/weight_methods.py:35
↓ 1 callers
Method
cagrad
(self, grads, alpha=0.5, rescale=1)
methods/weight_methods.py:576
↓ 1 callers
Function
delta_fn
(a)
experiments/nyuv2/utils.py:81
↓ 1 callers
Function
delta_fn
(a)
experiments/quantum_chemistry/utils.py:91
↓ 1 callers
Method
find_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 callers
Method
get_weighted_loss
( self, losses: torch.Tensor, shared_parameters: Union[List[torch.nn.parameter.Paramet
methods/weight_methods.py:22
↓ 1 callers
Method
get_weighted_loss
Parameters ---------- losses : shared_parameters : shared parameters kwargs : Returns -----
methods/weight_methods.py:190
↓ 1 callers
Method
get_weighted_loss
(self, losses: torch.Tensor, **kwargs)
methods/weight_methods.py:411
↓ 1 callers
Method
get_weighted_loss
Parameters ---------- losses : shared_parameters : shared parameters kwargs : Returns -------
methods/weight_methods.py:542
↓ 1 callers
Method
grad2vec
(shared_params, grads, grad_dims, task)
methods/weight_methods.py:612
↓ 1 callers
Function
gradient_normalizers
(grads, losses, normalization_type)
methods/min_norm_solvers.py:201
↓ 1 callers
Method
last_shared_parameters
Parameters of the last shared layer. Returns -------
experiments/nyuv2/models.py:480
↓ 1 callers
Method
last_shared_parameters
(self)
experiments/quantum_chemistry/models.py:63
↓ 1 callers
Function
main
(path, lr, bs, device)
experiments/nyuv2/trainer.py:52
↓ 1 callers
Function
main
( data_path: str, batch_size: int, device: torch.device, method: str, weight_method_params
experiments/quantum_chemistry/trainer.py:67
↓ 1 callers
Function
main
(method_type, device, n_iter, scale)
experiments/toy/trainer.py:23
↓ 1 callers
Method
overwrite_grad
(self, shared_parameters, newgrad, grad_dims)
methods/weight_methods.py:628
↓ 1 callers
Function
plot_2d_pareto
Adaptation of code from: https://github.com/Cranial-XIX/CAGrad
experiments/toy/utils.py:11
↓ 1 callers
Function
requirements
()
setup.py:14
↓ 1 callers
Method
shared_modules
(self)
experiments/nyuv2/models.py:114
↓ 1 callers
Method
shared_parameters
(self)
experiments/nyuv2/models.py:474
↓ 1 callers
Method
shared_parameters
(self)
experiments/quantum_chemistry/models.py:51
↓ 1 callers
Method
solve_optimization
(self, gtg: np.array)
methods/weight_methods.py:133
↓ 1 callers
Method
task_specific_parameters
(self)
experiments/nyuv2/models.py:477
↓ 1 callers
Method
task_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
Method
backward
( self, losses: torch.Tensor, shared_parameters: Union[ List[torch.nn.para
methods/weight_methods.py:246
Method
backward
( self, losses: torch.Tensor, parameters: Union[List[torch.nn.parameter.Parameter], to
methods/weight_methods.py:641
Function
count_parameters
(model)
experiments/utils.py:80
Method
find_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
Method
forward
(self, x)
experiments/nyuv2/models.py:181
Method
forward
(self, x, return_representation=False)
experiments/nyuv2/models.py:305
Method
forward
(self, x)
experiments/nyuv2/models.py:430
Method
forward
(self, x, return_representation=False)
experiments/nyuv2/models.py:487
Method
forward
(self, data, return_representation=False)
experiments/quantum_chemistry/models.py:33
Method
forward
(self, x, compute_grad=False)
experiments/toy/problem.py:14
Method
get_weighted_loss
(self, losses, **kwargs)
methods/weight_methods.py:292
Method
get_weighted_loss
(self, losses, **kwargs)
methods/weight_methods.py:314
Method
get_weighted_loss
Parameters ---------- losses : shared_parameters : last_shared_parameters : representation :
methods/weight_methods.py:349
Method
get_weighted_loss
(self, losses: torch.Tensor, **kwargs)
methods/weight_methods.py:427
Method
get_weighted_loss
( self, losses: torch.Tensor, shared_parameters: Union[ List[torch.nn.para
methods/weight_methods.py:460
Method
get_weighted_loss
(self, losses: torch.Tensor, **kwargs)
methods/weight_methods.py:663
next →
1–100 of 112, ranked by callers