Updates batch job configuration and model path

Updates the batch job configuration file to use the 'amdgpu' partition and increases the time limit.

Changes the model save path to a new versioned directory. This ensures that the training process saves the model to the correct location.

hoho_gpu_class.batch

@@ -3,10 +3,10 @@
 #SBATCH --ntasks-per-node=1 # 1 tasks per node
 #SBATCH --cpus-per-task=16 # 6 CPUS per task = 12 CPUS per node
 #SBATCH --mem-per-cpu=10G # 8GB per CPU = 96GB per node
-#SBATCH --time=4:00:00 # time limits: 1 hour
+#SBATCH --time=24:00:00 # time limits: 1 hour
 #SBATCH --error=hoho_gpu_class.err # standard error file
 #SBATCH --output=hoho_gpu_class.out # standard output file
-#SBATCH --partition=amdgpufast # partition name
+#SBATCH --partition=amdgpu # partition name
 #SBATCH --mail-user=skvrnjan@fel.cvut.cz # where send info about job
 #SBATCH --mail-type=ALL # what to send, valid type values are NONE, BEGIN, END, FAIL, REQUEUE, ALL
 #SBATCH --gres=gpu:1

train_pnet_class_cluster.py

@@ -5,7 +5,7 @@ if __name__ == "__main__":
 
     # Load the dataset
     dataset_path = "/mnt/personal/skvrnjan/hohocustom_edges/"
-    model_save_path = "/mnt/personal/skvrnjan/hoho_pnet_edges/initial.pth"
+    model_save_path = "/mnt/personal/skvrnjan/hoho_pnet_edges_v2/initial.pth"
 
     os.makedirs(model_save_path, exist_ok=True)