Skip to content

Conversation

@vict0rsch
Copy link
Collaborator

No description provided.

run_dir=config.get("run_dir", "./"),
is_debug=config.get("is_debug", False),
print_every=config.get("print_every", 100),
seed=config.get("seed", 0),
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pourquoi tu n'as pas mis tous les arguments du main ? Pour généraliser à FANet...

        trainer = registry.get_trainer_class(config.get("trainer", "energy"))(
            task=config["task"],
            model_attributes=config["model"],
            dataset=config["dataset"],
            optimizer=config["optim"],
            identifier=config["identifier"],
            timestamp_id=config.get("timestamp_id", None),
            run_dir=config.get("run_dir", "./"),
            is_debug=config.get("is_debug", False),
            print_every=config.get("print_every", 100),
            seed=config.get("seed", 0),
            logger=config.get("logger", "wandb"),
            local_rank=config["local_rank"],
            amp=config.get("amp", False),
            cpu=config.get("cpu", False),
            slurm=config.get("slurm", {}),
            new_gnn=config.get("new_gnn", True),
            frame_averaging=config.get("frame_averaging", None),
            data_split=config.get("data_split", None),
            note=config.get("note", ""),
            test_rotation_invariance=config.get("test_ri", None),
            choice_fa=config.get("choice_fa", None),
            wandb_tag=config.get("wandb_tag", None),
        ) 

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

parce que j'ai repris celui de get_data_sample.py mais t'as raison j'update

@vict0rsch vict0rsch merged commit 5eec5a2 into new-gnn Sep 16, 2022
@vict0rsch vict0rsch deleted the time-rewiring branch September 16, 2022 17:14
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants