PVNet 5.3.16__tar.gz → 5.3.18__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.
- {pvnet-5.3.16 → pvnet-5.3.18}/PKG-INFO +1 -1
- {pvnet-5.3.16 → pvnet-5.3.18}/PVNet.egg-info/PKG-INFO +1 -1
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/datamodule.py +11 -3
- {pvnet-5.3.16 → pvnet-5.3.18}/LICENSE +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/PVNet.egg-info/SOURCES.txt +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/PVNet.egg-info/dependency_links.txt +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/PVNet.egg-info/requires.txt +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/PVNet.egg-info/top_level.txt +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/README.md +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/__init__.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/load_model.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/__init__.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/base_model.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/ensemble.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/__init__.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/basic_blocks.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/encoders/__init__.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/encoders/basic_blocks.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/encoders/encoders3d.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/late_fusion.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/linear_networks/__init__.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/linear_networks/basic_blocks.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/linear_networks/networks.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/site_encoders/__init__.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/site_encoders/basic_blocks.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/models/late_fusion/site_encoders/encoders.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/optimizers.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/training/__init__.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/training/lightning_module.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/training/plots.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/training/train.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pvnet/utils.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/pyproject.toml +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/setup.cfg +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/tests/test_datamodule.py +0 -0
- {pvnet-5.3.16 → pvnet-5.3.18}/tests/test_end2end.py +0 -0
|
@@ -66,7 +66,6 @@ class PVNetDataModule(LightningDataModule):
|
|
|
66
66
|
num_workers=num_workers,
|
|
67
67
|
collate_fn=collate_fn,
|
|
68
68
|
pin_memory=pin_memory,
|
|
69
|
-
drop_last=False,
|
|
70
69
|
timeout=0,
|
|
71
70
|
worker_init_fn=None,
|
|
72
71
|
prefetch_factor=prefetch_factor,
|
|
@@ -121,8 +120,17 @@ class PVNetDataModule(LightningDataModule):
|
|
|
121
120
|
|
|
122
121
|
def train_dataloader(self) -> DataLoader:
|
|
123
122
|
"""Construct train dataloader"""
|
|
124
|
-
return DataLoader(
|
|
123
|
+
return DataLoader(
|
|
124
|
+
self.train_dataset,
|
|
125
|
+
shuffle=True,
|
|
126
|
+
drop_last=True,
|
|
127
|
+
**self._common_dataloader_kwargs
|
|
128
|
+
)
|
|
125
129
|
|
|
126
130
|
def val_dataloader(self) -> DataLoader:
|
|
127
131
|
"""Construct val dataloader"""
|
|
128
|
-
return DataLoader(
|
|
132
|
+
return DataLoader(
|
|
133
|
+
self.val_dataset,
|
|
134
|
+
shuffle=False,
|
|
135
|
+
drop_last=False,
|
|
136
|
+
**self._common_dataloader_kwargs)
|
|
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
|
|
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
|