We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0a10da2 commit 74aae2bCopy full SHA for 74aae2b
pvnet/models/multimodal/multimodal.py
@@ -343,7 +343,11 @@ def forward(self, x):
343
344
if self.include_sun:
345
sun = torch.cat(
346
- (x[BatchKey[f"{self.target_key_name}_solar_azimuth"]], x[BatchKey[f"{self.target_key_name}_solar_elevation"]]), dim=1
+ (
347
+ x[BatchKey[f"{self.target_key_name}_solar_azimuth"]],
348
+ x[BatchKey[f"{self.target_key_name}_solar_elevation"]],
349
+ ),
350
+ dim=1,
351
).float()
352
sun = self.sun_fc1(sun)
353
modes["sun"] = sun
0 commit comments