Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Feb 6, 2024
1 parent 6249e83 commit 68d209e
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion pvnet/models/multimodal/encoders/encoders3d.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,9 @@ def __init__(

# Calculate the size of the output of the 3D convolutional layers
cnn_output_size = conv3d_channels * cnn_spatial_output_size**2 * cnn_sequence_length
print(f"CNN sequence length: {cnn_sequence_length} and spatial output size: {cnn_spatial_output_size} for total output size: {cnn_output_size}")
print(
f"CNN sequence length: {cnn_sequence_length} and spatial output size: {cnn_spatial_output_size} for total output size: {cnn_output_size}"
)

self.final_block = nn.Sequential(
nn.Linear(in_features=cnn_output_size, out_features=fc_features),
Expand Down

0 comments on commit 68d209e

Please sign in to comment.