Update general.py

ultralytics · Feb 10, 2022 · 9e36b72 · 9e36b72
1 parent 244e6b7
commit 9e36b72
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/general.py b/utils/general.py
@@ -817,7 +817,7 @@ def print_mutation(results, hyp, save_dir, bucket):
                 f'# Last generation: {len(data) - 1}\n' +
                 '# ' + ', '.join(f'{x.strip():>20s}' for x in keys[:7]) + '\n' +
                 '# ' + ', '.join(f'{x:>20.5g}' for x in data.values[i, :7]) + '\n\n')
-        yaml.safe_dump(data.loc[i][7:], f, sort_keys=False)
+        yaml.safe_dump(data.loc[i][7:].to_dict(), f, sort_keys=False)
 
     if bucket:
         os.system(f'gsutil cp {evolve_csv} {evolve_yaml} gs://{bucket}')  # upload