PVNet 5.3.0__tar.gz → 5.3.2__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (36) hide show
  1. {pvnet-5.3.0 → pvnet-5.3.2}/PKG-INFO +1 -1
  2. {pvnet-5.3.0 → pvnet-5.3.2}/PVNet.egg-info/PKG-INFO +1 -1
  3. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/site_encoders/encoders.py +2 -2
  4. {pvnet-5.3.0 → pvnet-5.3.2}/LICENSE +0 -0
  5. {pvnet-5.3.0 → pvnet-5.3.2}/PVNet.egg-info/SOURCES.txt +0 -0
  6. {pvnet-5.3.0 → pvnet-5.3.2}/PVNet.egg-info/dependency_links.txt +0 -0
  7. {pvnet-5.3.0 → pvnet-5.3.2}/PVNet.egg-info/requires.txt +0 -0
  8. {pvnet-5.3.0 → pvnet-5.3.2}/PVNet.egg-info/top_level.txt +0 -0
  9. {pvnet-5.3.0 → pvnet-5.3.2}/README.md +0 -0
  10. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/__init__.py +0 -0
  11. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/datamodule.py +0 -0
  12. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/load_model.py +0 -0
  13. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/__init__.py +0 -0
  14. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/base_model.py +0 -0
  15. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/ensemble.py +0 -0
  16. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/__init__.py +0 -0
  17. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/basic_blocks.py +0 -0
  18. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/encoders/__init__.py +0 -0
  19. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/encoders/basic_blocks.py +0 -0
  20. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/encoders/encoders3d.py +0 -0
  21. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/late_fusion.py +0 -0
  22. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/linear_networks/__init__.py +0 -0
  23. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/linear_networks/basic_blocks.py +0 -0
  24. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/linear_networks/networks.py +0 -0
  25. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/site_encoders/__init__.py +0 -0
  26. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/models/late_fusion/site_encoders/basic_blocks.py +0 -0
  27. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/optimizers.py +0 -0
  28. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/training/__init__.py +0 -0
  29. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/training/lightning_module.py +0 -0
  30. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/training/plots.py +0 -0
  31. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/training/train.py +0 -0
  32. {pvnet-5.3.0 → pvnet-5.3.2}/pvnet/utils.py +0 -0
  33. {pvnet-5.3.0 → pvnet-5.3.2}/pyproject.toml +0 -0
  34. {pvnet-5.3.0 → pvnet-5.3.2}/setup.cfg +0 -0
  35. {pvnet-5.3.0 → pvnet-5.3.2}/tests/test_datamodule.py +0 -0
  36. {pvnet-5.3.0 → pvnet-5.3.2}/tests/test_end2end.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: PVNet
3
- Version: 5.3.0
3
+ Version: 5.3.2
4
4
  Summary: PVNet
5
5
  Author-email: Peter Dudfield <info@openclimatefix.org>
6
6
  Requires-Python: <3.14,>=3.11
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: PVNet
3
- Version: 5.3.0
3
+ Version: 5.3.2
4
4
  Summary: PVNet
5
5
  Author-email: Peter Dudfield <info@openclimatefix.org>
6
6
  Requires-Python: <3.14,>=3.11
@@ -158,7 +158,7 @@ class SingleAttentionNetwork(AbstractSitesEncoder):
158
158
  super().__init__(sequence_length, num_sites, out_features)
159
159
  self.sequence_length = sequence_length
160
160
  self.target_id_embedding = nn.Embedding(target_id_dim, out_features)
161
- self.id_embedding = nn.Embedding(num_sites, id_embed_dim)
161
+ self.site_id_embedding = nn.Embedding(num_sites, id_embed_dim)
162
162
  self._ids = nn.parameter.Parameter(torch.arange(num_sites), requires_grad=False)
163
163
  self.use_id_in_value = use_id_in_value
164
164
  self.key_to_use = key_to_use
@@ -224,7 +224,7 @@ class SingleAttentionNetwork(AbstractSitesEncoder):
224
224
  site_seqs, batch_size = self._encode_inputs(x)
225
225
 
226
226
  # site ID embeddings are the same for each sample
227
- id_embed = torch.tile(self.id_embedding(self._ids), (batch_size, 1, 1))
227
+ id_embed = torch.tile(self.site_id_embedding(self._ids), (batch_size, 1, 1))
228
228
  # Each concated (site sequence, site ID embedding) is processed with encoder
229
229
  x_seq_in = torch.cat((site_seqs, id_embed), dim=2).flatten(0, 1)
230
230
  key = self._key_encoder(x_seq_in)
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes