Adds slurm batch script and updates dataset path

Adds a slurm batch script for CPU training.

Updates the dataset directory path to a new location.
This improves data accessibility and management.

hoho_cpu.batch

@@ -0,0 +1,17 @@
+#!/bin/bash
+#SBATCH --nodes=1 # 1 node
+#SBATCH --ntasks-per-node=1 # 1 tasks per node
+#SBATCH --cpus-per-task=8 # 6 CPUS per task = 12 CPUS per node
+#SBATCH --mem-per-cpu=10G # 8GB per CPU = 96GB per node
+#SBATCH --time=24:00:00 # time limits: 1 hour
+#SBATCH --error=hoho_cpu.err # standard error file
+#SBATCH --output=hoho_cpu.out # standard output file
+#SBATCH --partition=amd # 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
+
+cd /mnt/personal/skvrnjan/hoho/
+module purge
+module load Python/3.10.8-GCCcore-12.2.0
+source /mnt/personal/skvrnjan/venvs/hoho/bin/activate
+python train.py

predict.py

@@ -14,7 +14,8 @@ import pyvista as pv
 from fast_pointnet import save_patches_dataset
 
 GENERATE_DATASET = True
-DATASET_DIR = '/home/skvrnjan/personal/hohocustom/'
+#DATASET_DIR = '/home/skvrnjan/personal/hohocustom/'
+DATASET_DIR = '/mnt/personal/skvrnjan/hohocustom'
 
 def convert_entry_to_human_readable(entry):
     out = {}