| """Default parameters for DGPS Gaussian pair experiments.""" | |
| import torch | |
| params = { | |
| "n": 1000, | |
| "μ_x": torch.tensor([0.0, 0.0]), | |
| "Σ_x": torch.tensor([[0.7, 0.5], [0.5, 0.7]]), | |
| "μ_y": torch.tensor([1.0, 1.0]), | |
| "Σ_y": torch.tensor([[0.4, -0.2], [-0.2, 0.4]]), | |
| } | |
| niters = 10000 | |
| model_size = 1000 | |
| inner_steps = 50 | |
| lr = 1e-2 | |
| inner_optimizer = "adam" | |
| batch_size = 512 | |