Lines Matching full:weight
44 weight = torch.tensor([[1.0]], device=device)
48 parameters = {f'{prefix}.l1.weight': weight,
52 parameters = {'l1.weight': weight,
58 prev_weight = to_check.l1.weight.clone()
62 # same as the input if the weight swapping went well.
65 # check that the weight remain unmodified
66 cur_weight = to_check.l1.weight
146 {'module.weight': torch.zeros(5, device='cuda')},
156 weight = torch.tensor([[1.0]], requires_grad=True)
159 parameters = {'l1.weight': weight,
165 self.assertIsNotNone(weight.grad)
169 self.assertIsNone(module.l1.weight.grad)
203 weight = torch.tensor([[1.0]])
206 parameters = {'l1.m.l1.weight': weight,
209 prev_weight = module.l1.weight.clone()
214 cur_weight = module.l1.weight
226 self.assertTrue('l1.parametrizations.weight.original' in dict(module.named_parameters()))
227 orig_sn_weight = module.l1.weight.clone()
232 … parameters = {'l1.parametrizations.weight.original': torch.nn.Parameter(torch.tensor([[1.0]])),
238 self.assertTrue('l1.parametrizations.weight.original' in dict(module.named_parameters()))
239 self.assertEqual(orig_sn_weight, module.l1.weight)
248 self.assertTrue('l1.parametrizations.weight.original' in dict(module.named_parameters()))
249 orig_sn_weight = module.l1.weight.clone()
253 … parameters = {'l1.parametrizations.weight.original': torch.nn.Parameter(torch.tensor([[1.0]])),
265 self.assertTrue('l1.parametrizations.weight.original' in dict(module.named_parameters()))
266 self.assertEqual(orig_sn_weight, module.l1.weight)
274 weight = torch.tensor([[2.0]])
279 parameters = {'l1.weight': weight}
282 self.assertEqual(out, x * weight + module.l1.bias + module.buffer)
284 parameters = {'l1.weight': weight,
288 self.assertEqual(out, x * weight + module.l1.bias + module.buffer)
296 weight = torch.tensor([[2.0]])
302 parameters = {'l1.weight': weight,
311 self.assertEqual(out, x * weight + bias + buffer)
314 parameters = {'l1.weight': weight}
327 parameters = {'l1.weight': weight,
343 parameters = {'l1.weight': weight,
366 weight = torch.tensor([[2.0]])
372 parameters = {'l1.weight': weight,
381 self.assertEqual(out, x * weight + buffer)
384 parameters = {'l1.weight': non_tensor}
397 parameters = {'l1.weight': weight, 'foo': torch.tensor([1.0])}
410 parameters = {'l1.weight': weight,
438 weight = torch.tensor([[2.0]])
443 parameters = {'l1.weight': weight,
448 self.assertEqual(out, x * weight + bias + bias + buffer + buffer)
450 parameters = {'l1.weight': weight,
456 self.assertEqual(out, x * weight + bias + bias + buffer + buffer)
464 weight = torch.tensor([[2.0]])
467 parameters = {'l1.weight': weight,
479 weight = torch.tensor([[1.0]])
483 parameters = {'l1.weight': weight,
513 weight = torch.tensor([[1.0]])
517 parameters = {'l1.weight': weight,
534 weight = torch.tensor([[2.0]])
540 parameters = {'l1.weight': weight,
549 self.assertEqual(out, x * weight + bias + bias + buffer + buffer)
552 parameters = {'l1.weight': weight,
567 parameters = {'l1.weight': weight,
581 parameters = {'l1.weight': weight,
597 parameters = {'l1.weight': weight,
613 parameters = {'l1.weight': weight,
757 … a = ({'l1.weight': torch.ones(1, 1)}, {'l1.bias': torch.ones(1)}, {'buffer': torch.zeros(1)})
764 parameters = {'l1.weight': torch.zeros((1, 1)), 'l1.bias': torch.zeros((1, 1))}
765 repeated_parameters = {'l1.weight': torch.ones((1, 1))}
768 re.escape("['l1.weight'] appeared in multiple dictionaries"),
789 weight = torch.tensor([[2.0]])
796 parameters = {'l1.weight': weight,
801 self.assertEqual(out, x * weight + bias + buffer)
802 self.assertEqual(parameters, (weight, bias))
804 self.assertTrue(all(t1 is t2 for t1, t2 in zip(parameters, (weight, bias))))
808 parameters = {'l1.weight': weight}
811 self.assertEqual(out, x * weight + module.l1.bias + module.buffer)
812 self.assertEqual(parameters, (weight, module.l1.bias))
814 self.assertTrue(all(t1 is t2 for t1, t2 in zip(parameters, (weight, module.l1.bias))))
818 parameters = {'l1.weight': weight,
824 self.assertEqual(out, x * weight + bias + buffer)
825 self.assertEqual(parameters, (weight, bias))
827 self.assertTrue(all(t1 is t2 for t1, t2 in zip(parameters, (weight, bias))))
831 parameters = {'l1.weight': weight,
837 self.assertEqual(out, x * weight + bias + buffer)
838 self.assertEqual(parameters, (weight, bias, extra_p))
840 self.assertTrue(all(t1 is t2 for t1, t2 in zip(parameters, (weight, bias, extra_p))))
844 parameters = {'l1.weight': weight,
848 self.assertEqual(out, x * weight + module.l1.bias + module.buffer)
849 self.assertEqual(parameters, (weight, module.l1.bias))
851 self.assertTrue(all(t1 is t2 for t1, t2 in zip(parameters, (weight, module.l1.bias))))
855 parameters = {'l1.weight': weight,
859 self.assertEqual(out, x * weight + module.l1.bias + module.buffer)
860 self.assertEqual(parameters, (weight, module.l1.bias, extra_p))
862 … self.assertTrue(all(t1 is t2 for t1, t2 in zip(parameters, (weight, module.l1.bias, extra_p))))
866 parameters = {'l1.weight': weight,
870 self.assertEqual(out, x * weight + module.buffer)
871 self.assertEqual(parameters, (weight,))
873 self.assertTrue(all(t1 is t2 for t1, t2 in zip(parameters, (weight,))))