(self, base)
| 52 | MAX_ROUGHNESS = 0.5 |
| 53 | |
| 54 | def __init__(self, base): |
| 55 | super(EnvironmentLight, self).__init__() |
| 56 | self.mtx = None |
| 57 | self.base = torch.nn.Parameter(base.clone().detach(), requires_grad=True) |
| 58 | self.register_parameter('env_base', self.base) |
| 59 | |
| 60 | def xfm(self, mtx): |
| 61 | self.mtx = mtx |