Skip to content
Snippets Groups Projects
Select Git revision
  • 5acac5442f4785deb046bc8dc2689ec67d630f12
  • 2023 default protected
2 results

jax.py

Blame
  • run_HPC.py NaN GiB
    __author__ = "Lukas Leufen"
    __date__ = '2020-06-29'
    
    import argparse
    from mlair.workflows import DefaultWorkflowHPC
    
    
    def main(parser_args):
    
        workflow = DefaultWorkflowHPC(**parser_args.__dict__)
        workflow.run()
    
    
    if __name__ == "__main__":
        parser = argparse.ArgumentParser()
        parser.add_argument('--experiment_date', metavar='--exp_date', type=str, default="testrun",
                            help="set experiment date as string")
        args = parser.parse_args()
        main(args)