[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
Clay-foundation · Mar 30, 2024 · a7778da · a7778da
1 parent af16073
commit a7778da
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/src/model_clay.py b/src/model_clay.py
@@ -1106,7 +1106,6 @@ def forward(self, x):
             encoded_unmasked_patches, unmasked_indices, masked_indices
         )
 
-
         return pixels
 
     def shared_step(self, batch, batch_idx):

diff --git a/trainer.py b/trainer.py
@@ -50,7 +50,7 @@ def cli_main(
     seed_everything_default=42,
     trainer_defaults: dict = {
         "accelerator": "auto",
-        "devices": "auto",#"strategy": "ddp_find_unused_parameters_true",
+        "devices": "auto",  # "strategy": "ddp_find_unused_parameters_true",
         "precision": "bf16-mixed",
         "log_every_n_steps": 1,
         "max_epochs": 10000,
-Original file line number
+Diff line change
@@ Expand Up / @@ -1106,7 +1106,6 @@ def forward(self, x): @@
                 encoded_unmasked_patches, unmasked_indices, masked_indices
             )
             return pixels
         def shared_step(self, batch, batch_idx):
@@ Expand Down @@