diff --git a/pufferlib/pufferl.py b/pufferlib/pufferl.py index 7132a19f9..8ba48a743 100644 --- a/pufferlib/pufferl.py +++ b/pufferlib/pufferl.py @@ -1160,6 +1160,10 @@ def export(args=None, env_name=None, vecenv=None, policy=None): weights = np.concatenate(weights) weights.tofile(path) print(f'Saved {len(weights)} weights to {path}') + + target_name = env_name.replace('puffer_', '') + print(f'You can copy this to the environment\'s resources using cp -L {path} ./resources/{target_name}/{target_name}_weights.bin') + os._exit(0) def autotune(args=None, env_name=None, vecenv=None, policy=None): package = args['package']