starting the job at: Mon Jun 2 09:24:01 CDT 2025 setting up the environment running the p1 categorization script cuda NVIDIA A100-PCIE-40GB _CudaDeviceProperties(name='NVIDIA A100-PCIE-40GB', major=8, minor=0, total_memory=40442MB, multi_processor_count=108, uuid=c91b110a-9eb1-15b6-ff0a-7aeb47b26ff0, L2_cache_size=40MB) Loading checkpoint shards: 0%| | 0/6 [00:00