Line too long (92 > 79 characters):
17 model = initialize_model(config, d_out=d_out, is_featurizer=False).to(config.device)Line too long (80 > 79 characters):
56 group_pairs.append((group_indices[group_i], group_indices[group_j]))Line too long (82 > 79 characters):
84 p_tgt.grad.data.add_(p_src.grad.data / n_groups_per_batch)Line too long (107 > 79 characters):
98 grad_inner_j = torch.autograd.grad(loss_inner_j, inner_net.parameters(), allow_unused=True)Line too long (95 > 79 characters):
109 p.grad.data.add_(self.config.mldg_beta * g_j.data / n_groups_per_batch)Line too long (89 > 79 characters):
130 return self.loss.compute(results['y_pred'], results['y_true'], return_dict=False)