diff --git a/main.py b/main.py index ad7cf296e25c576ed16341811d6edd8f8ecd4798..dedee957f94dd6cd4a20341633a8e30ee494726c 100644 --- a/main.py +++ b/main.py @@ -89,7 +89,7 @@ def sample_func(f): print(f"SAMPLING SETTINGS:\n\n {sampling_setting}\n\n") print("\n\nSTART SAMPLING\n\n") - globals()[meta_setting["sampling_function"]](model=framework,device=device,sample_all=sample_all, **sampling_setting,) + globals()[meta_setting["sampling_function"]](model=framework,device=device, **sampling_setting,) print("\n\nFINISHED SAMPLING\n\n")