Skip to content

Commit

Permalink
Add info-level logger message
Browse files Browse the repository at this point in the history
  • Loading branch information
tomaarsen committed Apr 25, 2024
1 parent bb0a139 commit a6d09f5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/integrations/integration_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -789,7 +789,7 @@ def setup(self, args, state, model, **kwargs):
try:
self._wandb.config["model/num_parameters"] = model.num_parameters()
except AttributeError:
pass
logger.info("Could not log the number of model parameters in Weights & Biases.")

# log the initial model and architecture to an artifact
with tempfile.TemporaryDirectory() as temp_dir:
Expand Down

0 comments on commit a6d09f5

Please sign in to comment.