| |
| import argparse, json, os |
| def main(): |
| parser = argparse.ArgumentParser() |
| parser.add_argument('--task', required=True) |
| parser.add_argument('--out_dir', default='experiments') |
| parser.add_argument('--use_wandb', action='store_true') |
| args = parser.parse_args() |
| os.makedirs(args.out_dir, exist_ok=True) |
| import numpy as np |
| phi = np.zeros((9,9)) |
| np.save(os.path.join(args.out_dir, f"{args.task}_phi_best.npy"), phi) |
| logs = [[{'accepted': True, 'atomic': '<Demo>', 'residue': 0, 'gates': {'A': True}}]] |
| with open(os.path.join(args.out_dir, f"{args.task}_logs.json"), 'w') as fh: |
| json.dump(logs, fh) |
| print("Wrote artifacts to", args.out_dir) |
| if __name__ == '__main__': |
| main() |
|
|
