Skip to content

Commit 55a69e1

Browse files
committed
Fix channel issue
1 parent ec277ae commit 55a69e1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pvnet/models/multimodal/site_encoders/encoders.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ def __init__(
129129
target_id_dim: int = 318,
130130
target_key_to_use: str = "gsp",
131131
input_key_to_use: str = "pv",
132-
num_channels: int = 0,
132+
num_channels: int = 1,
133133
num_sites_in_inference: int = 1,
134134
):
135135
"""A simple attention-based model with a single multihead attention layer
@@ -151,7 +151,7 @@ def __init__(
151151
target_id_dim: The number of unique IDs.
152152
target_key_to_use: The key to use for the target in the attention layer.
153153
input_key_to_use: The key to use for the input in the attention layer.
154-
num_channels: Number of channels in the input data. For single site generation, this will be 0, as there
154+
num_channels: Number of channels in the input data. For single site generation, this will be 1, as there
155155
is not channel dimension, for Sensors, this will probably be higher than that
156156
num_sites_in_inference: Number of sites to use in inference. This is used to determine the
157157
number of sites to use in the attention layer, for a single site, 1 works, while for multiple sites

0 commit comments

Comments
 (0)