Skip to content

Commit af3b19c

Browse files
committed
make no model upload default in sweep
1 parent be7254d commit af3b19c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pufferlib/pufferl.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1040,6 +1040,7 @@ def sweep(args=None, env_name=None):
10401040
args = args or load_config(env_name)
10411041
if not args['wandb'] and not args['neptune']:
10421042
raise pufferlib.APIUsageError('Sweeps require either wandb or neptune')
1043+
args['no_model_upload'] = True # Uploading trained model during sweep crashed wandb
10431044

10441045
method = args['sweep'].pop('method')
10451046
try:

0 commit comments

Comments
 (0)