Skip to content

Commit

Permalink
remove stray comments
Browse files Browse the repository at this point in the history
  • Loading branch information
smitchaudhary committed Jul 19, 2024
1 parent a0706bb commit aa9d9a2
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 8 deletions.
4 changes: 0 additions & 4 deletions qadence/ml_tools/train_grad.py
Original file line number Diff line number Diff line change
Expand Up @@ -137,10 +137,6 @@ def loss_fn(model: torch.nn.Module, data: torch.Tensor) -> tuple[torch.Tensor, d
else:
writer = importlib.import_module("mlflow")

# writer.mlflow.pytorch.autolog(
# log_every_n_step=config.write_every, log_models=False, log_datasets=False
# )

perform_val = isinstance(config.val_every, int)
if perform_val:
if not isinstance(dataloader, DictDataLoader):
Expand Down
4 changes: 0 additions & 4 deletions qadence/ml_tools/train_no_grad.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,10 +84,6 @@ def _update_parameters(
else:
writer = importlib.import_module("mlflow")

# writer.mlflow.pytorch.autolog(
# log_every_n_step=config.write_every, log_models=False, log_datasets=False
# )

# set optimizer configuration and initial parameters
optimizer.budget = config.max_iter
optimizer.enable_pickling()
Expand Down

0 comments on commit aa9d9a2

Please sign in to comment.