plot-components.py [30:42]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
theta = jnp.linspace(0, 2 * jnp.pi, 2 * NUM_POINTS)
phi = jnp.linspace(0, jnp.pi, NUM_POINTS)
tp = jnp.array(np.meshgrid(theta, phi, indexing='ij'))
tp = tp.transpose([1, 2, 0]).reshape(-1, 2)

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('exp_root', type=str)
    args = parser.parse_args()

    fname = f"{args.exp_root}/latest.pkl"
    with open(fname, 'rb') as f:
        W = pkl.load(f)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



plot-demo.py [32:44]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
theta = jnp.linspace(0, 2 * jnp.pi, 2 * NUM_POINTS)
phi = jnp.linspace(0, jnp.pi, NUM_POINTS)
tp = jnp.array(np.meshgrid(theta, phi, indexing='ij'))
tp = tp.transpose([1, 2, 0]).reshape(-1, 2)

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('exp_root', type=str)
    args = parser.parse_args()

    fname = f"{args.exp_root}/latest.pkl"
    with open(fname, 'rb') as f:
        W = pkl.load(f)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -