Skip to content

Commit

Permalink
remove parent
Browse files Browse the repository at this point in the history
  • Loading branch information
misko committed Aug 21, 2024
1 parent 0b4c5ee commit 07efac0
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions src/fairchem/core/trainers/base_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -127,7 +127,6 @@ def __init__(
"gpus": distutils.get_world_size() if not self.cpu else 0,
"cmd": {
"identifier": identifier,
"parent": identifier,
"print_every": print_every,
"seed": seed,
"timestamp_id": self.timestamp_id,
Expand Down Expand Up @@ -575,9 +574,6 @@ def load_checkpoint(
self.step = checkpoint.get("step", 0)
self.best_val_metric = checkpoint.get("best_val_metric", None)
self.primary_metric = checkpoint.get("primary_metric", None)
self.config["cmd"]["parent"] = checkpoint["config"]["cmd"].get(
"parent", "identifier"
)

new_dict = match_state_dict(self.model.state_dict(), checkpoint["state_dict"])
strict = self.config.get("task", {}).get("strict_load", True)
Expand Down

0 comments on commit 07efac0

Please sign in to comment.