(self, input_dir)
| 104 | json.dump(self.get_args_for_saving(), f) |
| 105 | |
| 106 | def load(self, input_dir): |
| 107 | if input_dir: |
| 108 | model_args_file = os.path.join(input_dir, "model_args.json") |
| 109 | if os.path.isfile(model_args_file): |
| 110 | with open(model_args_file, "r") as f: |
| 111 | model_args = json.load(f) |
| 112 | |
| 113 | self.update_from_dict(model_args) |
| 114 | |
| 115 | |
| 116 | @dataclass |