From c6c00517244205257df69a6136fe8d72d3bde7e6 Mon Sep 17 00:00:00 2001 From: JamesTrewern <trewern.james@gmail.com> Date: Wed, 12 Apr 2023 22:36:50 +0100 Subject: [PATCH] added if name == main guard for python multi threading on dataloaders --- main.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 3b9fcebb..2f39a123 100644 --- a/main.py +++ b/main.py @@ -2,11 +2,12 @@ import datahandler import os from constants import EPOCHS -print(EPOCHS) - -df_ham_train, df_ham_val = datahandler.getHamDataLoaders() -#print(df_ham_train) -#print(df_ham_val) +def main(): + df_ham_train, df_ham_val = datahandler.getHamDataLoaders() + #print(df_ham_train) + #print(df_ham_val) +if __name__ == '__main__': + main() #FYI training samples for all but the 'melanocytic nevi' have been equalised (samples duplicated) so they're an even distribution # Before and after prints of the tables are uncommented to show (lines 142 and 154 in datahandler) \ No newline at end of file -- GitLab