Lmod Warning:
-------------------------------------------------------------------------------
The following dependent module(s) are not currently loaded: curl/8.4.0
(required by: htslib/1.16)
-------------------------------------------------------------------------------




The following have been reloaded with a version change:
  1) curl/8.4.0 => curl/8.17.0     2) openssl/3.0.7 => openssl/3.6.0

2025-11-17 18:26:35.269623: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:26:49.039280: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-11-17 18:26:49.040625: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2025-11-17 18:26:49.066997: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2025-11-17 18:26:49.067047: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:26:49.071396: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-11-17 18:26:49.071461: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-11-17 18:26:49.073471: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-11-17 18:26:49.074710: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-11-17 18:26:49.077953: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-11-17 18:26:49.079409: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-11-17 18:26:49.080473: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-11-17 18:26:49.080860: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-11-17 18:26:49.081266: I tensorflow/core/platform/cpu_feature_guard.cc:142] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 AVX512F FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2025-11-17 18:26:49.082957: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-11-17 18:26:49.083217: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2025-11-17 18:26:49.083239: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:26:49.083253: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-11-17 18:26:49.083263: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-11-17 18:26:49.083272: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-11-17 18:26:49.083281: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-11-17 18:26:49.083291: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-11-17 18:26:49.083300: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-11-17 18:26:49.083321: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-11-17 18:26:49.083611: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-11-17 18:26:49.083631: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:26:49.494185: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2025-11-17 18:26:49.494281: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2025-11-17 18:26:49.494293: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2025-11-17 18:26:49.495025: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 42313 MB memory) -> physical GPU (device: 0, name: NVIDIA L40S, pci bus id: 0000:61:00.0, compute capability: 8.9)
2025-11-17 18:26:58.111709: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2025-11-17 18:26:58.112367: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2500000000 Hz
2025-11-17 18:26:59.275104: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-11-17 18:26:59.774305: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-11-17 18:26:59.780523: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-11-17 18:27:01.289810: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2025-11-17 18:27:01.410067: W tensorflow/stream_executor/gpu/redzone_allocator.cc:314] Internal: ptxas exited with non-zero error code 256, output: 
Relying on driver to perform ptx compilation. 
Modify $PATH to customize ptxas location.
This message will be only logged once.
2025-11-17 18:27:24.155249: I tensorflow/stream_executor/cuda/cuda_blas.cc:1838] TensorFloat-32 will be used for the matrix multiplication. This will only be logged once.
/home/users/shouvikm/miniconda3/envs/bpnet/lib/python3.7/site-packages/tensorflow/python/keras/engine/functional.py:595: UserWarning: Input dict contained keys ['coordinates', 'jitters', 'index', 'status', 'rev_comp'] which did not match any model input. They will be ignored by the model.
  [n for n in tensors.keys() if n not in ref_input_names])
/home/users/shouvikm/miniconda3/envs/bpnet/lib/python3.7/site-packages/tensorflow/python/keras/engine/functional.py:595: UserWarning: Input dict contained keys ['coordinates'] which did not match any model input. They will be ignored by the model.
  [n for n in tensors.keys() if n not in ref_input_names])
2025-11-17 18:34:27.502922: W tensorflow/python/util/util.cc:348] Sets are not currently considered sequences, but this may change in the future, so consider avoiding using them.
2025-11-17 18:34:31.200240: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:34:36.681059: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-11-17 18:34:36.681866: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2025-11-17 18:34:36.708484: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2025-11-17 18:34:36.708533: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:34:36.712680: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-11-17 18:34:36.712764: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-11-17 18:34:36.714768: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-11-17 18:34:36.715885: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-11-17 18:34:36.719031: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-11-17 18:34:36.720465: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-11-17 18:34:36.721626: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-11-17 18:34:36.722009: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-11-17 18:34:36.722245: I tensorflow/core/platform/cpu_feature_guard.cc:142] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 AVX512F FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2025-11-17 18:34:36.723658: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-11-17 18:34:36.723908: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2025-11-17 18:34:36.723930: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:34:36.723944: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-11-17 18:34:36.723954: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-11-17 18:34:36.723964: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-11-17 18:34:36.723974: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-11-17 18:34:36.723984: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-11-17 18:34:36.723993: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-11-17 18:34:36.724003: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-11-17 18:34:36.724286: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-11-17 18:34:36.724307: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:34:37.137065: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2025-11-17 18:34:37.137159: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2025-11-17 18:34:37.137170: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2025-11-17 18:34:37.137859: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 42313 MB memory) -> physical GPU (device: 0, name: NVIDIA L40S, pci bus id: 0000:61:00.0, compute capability: 8.9)
/home/users/shouvikm/miniconda3/envs/bpnet/lib/python3.7/site-packages/tensorflow/python/keras/layers/core.py:1059: UserWarning: bpnet.model.arch is not loaded, but a Lambda layer uses it. It may cause errors.
  , UserWarning)

batch:   0%|          | 0/92 [00:00<?, ?it/s]2025-11-17 18:34:38.740337: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2025-11-17 18:34:38.740897: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2500000000 Hz
2025-11-17 18:34:38.938631: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-11-17 18:34:39.453531: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-11-17 18:34:39.455552: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-11-17 18:34:41.057410: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2025-11-17 18:34:41.170981: W tensorflow/stream_executor/gpu/redzone_allocator.cc:314] Internal: ptxas exited with non-zero error code 256, output: 
Relying on driver to perform ptx compilation. 
Modify $PATH to customize ptxas location.
This message will be only logged once.
/home/users/shouvikm/miniconda3/envs/bpnet/lib/python3.7/site-packages/tensorflow/python/keras/engine/functional.py:595: UserWarning: Input dict contained keys ['coordinates', 'true_profiles', 'true_logcounts', 'rev_comp'] which did not match any model input. They will be ignored by the model.
  [n for n in tensors.keys() if n not in ref_input_names])

batch:   1%|          | 1/92 [00:29<45:29, 29.99s/it]
batch:   2%|▏         | 2/92 [00:30<18:44, 12.50s/it]
batch:   3%|▎         | 3/92 [00:30<10:14,  6.90s/it]
batch:   4%|▍         | 4/92 [00:30<06:15,  4.27s/it]
batch:   5%|▌         | 5/92 [00:30<04:04,  2.81s/it]
batch:   7%|▋         | 6/92 [00:31<02:46,  1.94s/it]
batch:   8%|▊         | 7/92 [00:31<01:57,  1.38s/it]
batch:   9%|▊         | 8/92 [00:31<01:25,  1.01s/it]
batch:  10%|▉         | 9/92 [00:31<01:03,  1.30it/s]
batch:  11%|█         | 10/92 [00:32<00:49,  1.66it/s]
batch:  12%|█▏        | 11/92 [00:32<00:39,  2.04it/s]
batch:  13%|█▎        | 12/92 [00:32<00:32,  2.42it/s]
batch:  14%|█▍        | 13/92 [00:32<00:28,  2.79it/s]
batch:  15%|█▌        | 14/92 [00:33<00:25,  3.12it/s]
batch:  16%|█▋        | 15/92 [00:33<00:22,  3.37it/s]
batch:  17%|█▋        | 16/92 [00:33<00:21,  3.60it/s]
batch:  18%|█▊        | 17/92 [00:33<00:19,  3.76it/s]
batch:  20%|█▉        | 18/92 [00:33<00:19,  3.89it/s]
batch:  21%|██        | 19/92 [00:34<00:18,  3.99it/s]
batch:  22%|██▏       | 20/92 [00:34<00:17,  4.06it/s]
batch:  23%|██▎       | 21/92 [00:34<00:17,  4.13it/s]
batch:  24%|██▍       | 22/92 [00:34<00:16,  4.16it/s]
batch:  25%|██▌       | 23/92 [00:35<00:16,  4.16it/s]
batch:  26%|██▌       | 24/92 [00:35<00:16,  4.18it/s]
batch:  27%|██▋       | 25/92 [00:35<00:16,  4.18it/s]
batch:  28%|██▊       | 26/92 [00:35<00:15,  4.13it/s]
batch:  29%|██▉       | 27/92 [00:36<00:15,  4.14it/s]
batch:  30%|███       | 28/92 [00:36<00:15,  4.18it/s]
batch:  32%|███▏      | 29/92 [00:36<00:15,  4.20it/s]
batch:  33%|███▎      | 30/92 [00:36<00:14,  4.20it/s]
batch:  34%|███▎      | 31/92 [00:37<00:14,  4.17it/s]
batch:  35%|███▍      | 32/92 [00:37<00:14,  4.18it/s]
batch:  36%|███▌      | 33/92 [00:37<00:14,  4.21it/s]
batch:  37%|███▋      | 34/92 [00:37<00:13,  4.21it/s]
batch:  38%|███▊      | 35/92 [00:38<00:13,  4.24it/s]
batch:  39%|███▉      | 36/92 [00:38<00:13,  4.24it/s]
batch:  40%|████      | 37/92 [00:38<00:12,  4.23it/s]
batch:  41%|████▏     | 38/92 [00:38<00:12,  4.22it/s]
batch:  42%|████▏     | 39/92 [00:38<00:12,  4.24it/s]
batch:  43%|████▎     | 40/92 [00:39<00:12,  4.26it/s]
batch:  45%|████▍     | 41/92 [00:39<00:11,  4.28it/s]
batch:  46%|████▌     | 42/92 [00:39<00:11,  4.28it/s]
batch:  47%|████▋     | 43/92 [00:39<00:11,  4.30it/s]
batch:  48%|████▊     | 44/92 [00:40<00:11,  4.30it/s]
batch:  49%|████▉     | 45/92 [00:40<00:10,  4.31it/s]
batch:  50%|█████     | 46/92 [00:40<00:10,  4.31it/s]
batch:  51%|█████     | 47/92 [00:40<00:10,  4.31it/s]
batch:  52%|█████▏    | 48/92 [00:41<00:10,  4.31it/s]
batch:  53%|█████▎    | 49/92 [00:41<00:10,  4.26it/s]
batch:  54%|█████▍    | 50/92 [00:41<00:09,  4.24it/s]
batch:  55%|█████▌    | 51/92 [00:41<00:09,  4.25it/s]
batch:  57%|█████▋    | 52/92 [00:42<00:09,  4.27it/s]
batch:  58%|█████▊    | 53/92 [00:42<00:09,  4.28it/s]
batch:  59%|█████▊    | 54/92 [00:42<00:08,  4.28it/s]
batch:  60%|█████▉    | 55/92 [00:42<00:08,  4.30it/s]
batch:  61%|██████    | 56/92 [00:42<00:08,  4.31it/s]
batch:  62%|██████▏   | 57/92 [00:43<00:08,  4.31it/s]
batch:  63%|██████▎   | 58/92 [00:43<00:07,  4.30it/s]
batch:  64%|██████▍   | 59/92 [00:43<00:07,  4.31it/s]
batch:  65%|██████▌   | 60/92 [00:43<00:07,  4.30it/s]
batch:  66%|██████▋   | 61/92 [00:44<00:07,  4.28it/s]
batch:  67%|██████▋   | 62/92 [00:44<00:06,  4.29it/s]
batch:  68%|██████▊   | 63/92 [00:44<00:06,  4.29it/s]
batch:  70%|██████▉   | 64/92 [00:44<00:06,  4.30it/s]
batch:  71%|███████   | 65/92 [00:45<00:06,  4.31it/s]
batch:  72%|███████▏  | 66/92 [00:45<00:07,  3.65it/s]
batch:  73%|███████▎  | 67/92 [00:45<00:06,  3.81it/s]
batch:  74%|███████▍  | 68/92 [00:45<00:06,  3.92it/s]
batch:  75%|███████▌  | 69/92 [00:46<00:05,  4.02it/s]
batch:  76%|███████▌  | 70/92 [00:46<00:05,  4.08it/s]
batch:  77%|███████▋  | 71/92 [00:46<00:05,  4.13it/s]
batch:  78%|███████▊  | 72/92 [00:46<00:04,  4.18it/s]
batch:  79%|███████▉  | 73/92 [00:47<00:04,  4.21it/s]
batch:  80%|████████  | 74/92 [00:47<00:04,  4.24it/s]
batch:  82%|████████▏ | 75/92 [00:47<00:03,  4.26it/s]
batch:  83%|████████▎ | 76/92 [00:47<00:03,  4.26it/s]
batch:  84%|████████▎ | 77/92 [00:47<00:03,  4.28it/s]
batch:  85%|████████▍ | 78/92 [00:48<00:03,  4.28it/s]
batch:  86%|████████▌ | 79/92 [00:48<00:03,  4.29it/s]
batch:  87%|████████▋ | 80/92 [00:48<00:02,  4.29it/s]
batch:  88%|████████▊ | 81/92 [00:48<00:02,  4.28it/s]
batch:  89%|████████▉ | 82/92 [00:49<00:02,  4.25it/s]
batch:  90%|█████████ | 83/92 [00:49<00:02,  4.24it/s]
batch:  91%|█████████▏| 84/92 [00:49<00:01,  4.26it/s]
batch:  92%|█████████▏| 85/92 [00:49<00:01,  4.24it/s]
batch:  93%|█████████▎| 86/92 [00:50<00:01,  4.25it/s]
batch:  95%|█████████▍| 87/92 [00:50<00:01,  4.25it/s]
batch:  96%|█████████▌| 88/92 [00:50<00:00,  4.25it/s]
batch:  97%|█████████▋| 89/92 [00:50<00:00,  4.26it/s]
batch:  98%|█████████▊| 90/92 [00:50<00:00,  4.50it/s]
batch:  99%|█████████▉| 91/92 [00:51<00:00,  4.44it/s]
batch: 100%|██████████| 92/92 [00:51<00:00,  1.79it/s]

  0%|          | 0/5811 [00:00<?, ?it/s]
  2%|▏         | 119/5811 [00:00<00:04, 1181.32it/s]
  4%|▍         | 240/5811 [00:00<00:04, 1197.64it/s]
  6%|▌         | 360/5811 [00:00<00:04, 1175.18it/s]
  8%|▊         | 480/5811 [00:00<00:04, 1184.27it/s]
 10%|█         | 601/5811 [00:00<00:04, 1188.56it/s]
 12%|█▏        | 720/5811 [00:00<00:04, 1178.77it/s]
 14%|█▍        | 838/5811 [00:00<00:04, 1178.92it/s]
 16%|█▋        | 956/5811 [00:00<00:04, 1175.21it/s]
 18%|█▊        | 1074/5811 [00:00<00:04, 1170.30it/s]
 21%|██        | 1192/5811 [00:01<00:03, 1161.18it/s]
 23%|██▎       | 1309/5811 [00:01<00:03, 1136.45it/s]
 24%|██▍       | 1423/5811 [00:01<00:03, 1137.46it/s]
 27%|██▋       | 1540/5811 [00:01<00:03, 1141.12it/s]
 29%|██▊       | 1657/5811 [00:01<00:03, 1143.76it/s]
 30%|███       | 1772/5811 [00:01<00:03, 1142.64it/s]
 32%|███▏      | 1887/5811 [00:01<00:03, 1133.81it/s]
 34%|███▍      | 2001/5811 [00:01<00:03, 1129.74it/s]
 36%|███▋      | 2116/5811 [00:01<00:03, 1129.94it/s]
 38%|███▊      | 2231/5811 [00:01<00:03, 1131.88it/s]
 40%|████      | 2345/5811 [00:02<00:03, 1132.91it/s]
 42%|████▏     | 2459/5811 [00:02<00:02, 1123.19it/s]
 44%|████▍     | 2573/5811 [00:02<00:02, 1122.82it/s]
 46%|████▌     | 2686/5811 [00:02<00:02, 1114.26it/s]
 48%|████▊     | 2798/5811 [00:02<00:02, 1113.21it/s]
 50%|█████     | 2910/5811 [00:02<00:02, 1111.65it/s]
 52%|█████▏    | 3022/5811 [00:02<00:02, 1105.25it/s]
 54%|█████▍    | 3133/5811 [00:02<00:02, 1100.42it/s]
 56%|█████▌    | 3244/5811 [00:02<00:02, 1099.58it/s]
 58%|█████▊    | 3354/5811 [00:02<00:02, 1094.25it/s]
 60%|█████▉    | 3466/5811 [00:03<00:02, 1100.67it/s]
 62%|██████▏   | 3577/5811 [00:03<00:02, 1098.55it/s]
 63%|██████▎   | 3687/5811 [00:03<00:01, 1084.89it/s]
 66%|██████▌   | 3807/5811 [00:03<00:01, 1115.21it/s]
 68%|██████▊   | 3924/5811 [00:03<00:01, 1128.30it/s]
 69%|██████▉   | 4037/5811 [00:03<00:01, 1106.68it/s]
 71%|███████▏  | 4148/5811 [00:03<00:01, 1097.87it/s]
 73%|███████▎  | 4258/5811 [00:03<00:01, 1064.38it/s]
 75%|███████▌  | 4366/5811 [00:03<00:01, 1068.14it/s]
 77%|███████▋  | 4473/5811 [00:03<00:01, 1066.95it/s]
 79%|███████▉  | 4581/5811 [00:04<00:01, 1067.81it/s]
 81%|████████  | 4688/5811 [00:04<00:01, 1064.07it/s]
 83%|████████▎ | 4795/5811 [00:04<00:00, 1050.85it/s]
 84%|████████▍ | 4902/5811 [00:04<00:00, 1053.90it/s]
 86%|████████▌ | 5008/5811 [00:04<00:00, 1039.04it/s]
 88%|████████▊ | 5113/5811 [00:04<00:00, 1041.46it/s]
 90%|████████▉ | 5220/5811 [00:04<00:00, 1049.60it/s]
 92%|█████████▏| 5326/5811 [00:04<00:00, 1045.80it/s]
 93%|█████████▎| 5431/5811 [00:04<00:00, 1046.22it/s]
 95%|█████████▌| 5536/5811 [00:05<00:00, 1041.11it/s]
 97%|█████████▋| 5641/5811 [00:05<00:00, 1039.17it/s]
 99%|█████████▉| 5745/5811 [00:05<00:00, 1035.58it/s]
100%|██████████| 5811/5811 [00:05<00:00, 1103.02it/s]

  0%|          | 0/5811 [00:00<?, ?it/s]
  2%|▏         | 116/5811 [00:00<00:04, 1157.51it/s]
  4%|▍         | 236/5811 [00:00<00:04, 1172.19it/s]
  6%|▌         | 355/5811 [00:00<00:04, 1180.04it/s]
  8%|▊         | 474/5811 [00:00<00:04, 1145.69it/s]
 10%|█         | 595/5811 [00:00<00:04, 1161.22it/s]
 12%|█▏        | 715/5811 [00:00<00:04, 1170.75it/s]
 14%|█▍        | 833/5811 [00:00<00:04, 1141.23it/s]
 16%|█▋        | 951/5811 [00:00<00:04, 1147.29it/s]
 18%|█▊        | 1070/5811 [00:00<00:04, 1155.35it/s]
 20%|██        | 1187/5811 [00:01<00:03, 1157.65it/s]
 22%|██▏       | 1303/5811 [00:01<00:03, 1147.52it/s]
 24%|██▍       | 1419/5811 [00:01<00:03, 1151.12it/s]
 26%|██▋       | 1535/5811 [00:01<00:03, 1127.36it/s]
 28%|██▊       | 1652/5811 [00:01<00:03, 1135.90it/s]
 30%|███       | 1767/5811 [00:01<00:03, 1134.66it/s]
 32%|███▏      | 1881/5811 [00:01<00:03, 1128.68it/s]
 34%|███▍      | 1994/5811 [00:01<00:03, 1126.38it/s]
 36%|███▋      | 2108/5811 [00:01<00:03, 1125.45it/s]
 38%|███▊      | 2223/5811 [00:01<00:03, 1130.23it/s]
 40%|████      | 2337/5811 [00:02<00:03, 1127.45it/s]
 42%|████▏     | 2450/5811 [00:02<00:03, 1120.05it/s]
 44%|████▍     | 2563/5811 [00:02<00:02, 1095.86it/s]
 46%|████▌     | 2675/5811 [00:02<00:02, 1099.30it/s]
 48%|████▊     | 2788/5811 [00:02<00:02, 1108.03it/s]
 50%|████▉     | 2899/5811 [00:02<00:02, 1107.07it/s]
 52%|█████▏    | 3010/5811 [00:02<00:02, 1075.78it/s]
 54%|█████▎    | 3121/5811 [00:02<00:02, 1085.08it/s]
 56%|█████▌    | 3232/5811 [00:02<00:02, 1091.16it/s]
 58%|█████▊    | 3344/5811 [00:02<00:02, 1098.74it/s]
 59%|█████▉    | 3455/5811 [00:03<00:02, 1095.69it/s]
 61%|██████▏   | 3565/5811 [00:03<00:02, 1095.63it/s]
 63%|██████▎   | 3677/5811 [00:03<00:01, 1099.01it/s]
 65%|██████▌   | 3787/5811 [00:03<00:01, 1094.29it/s]
 67%|██████▋   | 3911/5811 [00:03<00:01, 1134.66it/s]
 69%|██████▉   | 4025/5811 [00:03<00:01, 1120.00it/s]
 71%|███████   | 4138/5811 [00:03<00:01, 1104.58it/s]
 73%|███████▎  | 4249/5811 [00:03<00:01, 1097.79it/s]
 75%|███████▌  | 4359/5811 [00:03<00:01, 1092.15it/s]
 77%|███████▋  | 4469/5811 [00:03<00:01, 1087.33it/s]
 79%|███████▉  | 4578/5811 [00:04<00:01, 1080.72it/s]
 81%|████████  | 4687/5811 [00:04<00:01, 1078.17it/s]
 83%|████████▎ | 4795/5811 [00:04<00:00, 1061.45it/s]
 84%|████████▍ | 4902/5811 [00:04<00:00, 1062.26it/s]
 86%|████████▌ | 5010/5811 [00:04<00:00, 1062.40it/s]
 88%|████████▊ | 5117/5811 [00:04<00:00, 1034.95it/s]
 90%|████████▉ | 5225/5811 [00:04<00:00, 1042.09it/s]
 92%|█████████▏| 5331/5811 [00:04<00:00, 1042.15it/s]
 94%|█████████▎| 5438/5811 [00:04<00:00, 1048.09it/s]
 95%|█████████▌| 5544/5811 [00:05<00:00, 1046.12it/s]
 97%|█████████▋| 5649/5811 [00:05<00:00, 1045.15it/s]
 99%|█████████▉| 5754/5811 [00:05<00:00, 1018.76it/s]
100%|██████████| 5811/5811 [00:05<00:00, 1099.39it/s]
2025-11-17 18:35:53.640958: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:35:59.311737: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-11-17 18:35:59.312928: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2025-11-17 18:35:59.340520: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2025-11-17 18:35:59.340584: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:35:59.344861: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-11-17 18:35:59.344945: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-11-17 18:35:59.347052: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-11-17 18:35:59.348357: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-11-17 18:35:59.351521: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-11-17 18:35:59.353037: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-11-17 18:35:59.354161: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-11-17 18:35:59.354562: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-11-17 18:35:59.354908: I tensorflow/core/platform/cpu_feature_guard.cc:142] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2 AVX512F FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2025-11-17 18:35:59.356426: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-11-17 18:35:59.356708: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2025-11-17 18:35:59.356741: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:35:59.356757: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-11-17 18:35:59.356770: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-11-17 18:35:59.356783: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-11-17 18:35:59.356796: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-11-17 18:35:59.356809: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-11-17 18:35:59.356822: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-11-17 18:35:59.356835: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-11-17 18:35:59.357134: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-11-17 18:35:59.357160: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-11-17 18:35:59.768165: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2025-11-17 18:35:59.768260: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2025-11-17 18:35:59.768269: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2025-11-17 18:35:59.769001: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 42313 MB memory) -> physical GPU (device: 0, name: NVIDIA L40S, pci bus id: 0000:61:00.0, compute capability: 8.9)
2025-11-17 18:35:59.809182: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:196] None of the MLIR optimization passes are enabled (registered 0 passes)
2025-11-17 18:35:59.822800: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2500000000 Hz
2025-11-17 18:36:04.528681: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-11-17 18:36:05.040147: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-11-17 18:36:05.044707: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-11-17 18:36:06.608390: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2025-11-17 18:36:06.739604: W tensorflow/stream_executor/gpu/redzone_allocator.cc:314] Internal: ptxas exited with non-zero error code 256, output: 
Relying on driver to perform ptx compilation. 
Modify $PATH to customize ptxas location.
This message will be only logged once.
2025-11-17 18:36:27.622766: I tensorflow/stream_executor/cuda/cuda_blas.cc:1838] TensorFloat-32 will be used for the matrix multiplication. This will only be logged once.
RuntimeError: module compiled against API version 0xe but this version of numpy is 0xd
/home/users/shouvikm/miniconda3/envs/bpnet/lib/python3.7/site-packages/tensorflow/python/keras/layers/core.py:1059: UserWarning: bpnet.model.arch is not loaded, but a Lambda layer uses it. It may cause errors.
  , UserWarning)