From 012ad7791e5befcc38e3379ba94c9dffcfad7725 Mon Sep 17 00:00:00 2001 From: "Dr. Mahdi Hashemzadeh" <59253242+M-Hashemzadeh@users.noreply.github.com> Date: Sat, 31 Aug 2024 13:34:50 +0330 Subject: [PATCH] Update UFS_Net_train.py --- UFS_Net_train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/UFS_Net_train.py b/UFS_Net_train.py index 35e9b82..cc3b0ca 100644 --- a/UFS_Net_train.py +++ b/UFS_Net_train.py @@ -17,6 +17,7 @@ import copy from UFS_Net_architecture import UFSNet + batch_size = 64 # Define 8 class labels