modified batch script and training script

.gitignore

code/__pycache__/model.cpython-38.pyc

code/__pycache__/utils.cpython-38.pyc

code/profiler/ggpu147_10463.1679675485098.pt.trace.json

code/profiler/ggpu147_10463.1679675494591.pt.trace.json

code/profiler/ggpu147_9810.1679675095745.pt.trace.json

code/profiler/ggpu147_9810.1679675105186.pt.trace.json

code/requirements.txt

code/slurm-train-nn-gpu-4448294.out

code/slurm_files/slurm-train-nn-gpu-4448370.err

code/slurm_files/slurm-train-nn-gpu-4448370.out

code/submit_train.sh

...nsorboard/events.out.tfevents.1679664823.ggpu202.103679.0

...tensorboard/events.out.tfevents.1679675078.ggpu147.9810.0

...ensorboard/events.out.tfevents.1679675467.ggpu147.10463.0

code/train_with_logger.py