Skip to content

Commit c52e86c

Browse files
committed
Merge branch 'impswp' into g2048-65k
2 parents 91e372a + af3b19c commit c52e86c

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)