File size: 750 Bytes
b48dd06 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | #!/usr/bin/env python3
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()
|