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

2026-05-28 11:39:54.316808: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:41:43.481043: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-05-28 11:41:43.505034: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-05-28 11:41:43.552231: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:4a:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-05-28 11:41:43.552322: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:41:44.216662: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:41:44.216810: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:41:44.536804: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:41:45.134780: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:41:45.721129: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:41:45.951411: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:41:46.193798: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:41:46.194597: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:41:46.195071: 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.
2026-05-28 11:41:46.195259: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-05-28 11:41:46.195661: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:4a:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-05-28 11:41:46.195685: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:41:46.195700: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:41:46.195710: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:41:46.195720: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:41:46.195729: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:41:46.195738: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:41:46.195748: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:41:46.195776: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:41:46.196210: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:41:46.196235: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:41:48.761915: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-05-28 11:41:48.762015: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-05-28 11:41:48.762027: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-05-28 11:41:48.762689: 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:4a:00.0, compute capability: 8.9)
2026-05-28 11:41:58.162896: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2026-05-28 11:41:58.163545: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2500000000 Hz
2026-05-28 11:42:01.436234: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:42:02.862420: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:42:02.881056: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:42:18.851943: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-05-28 11:42:18.946173: 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.
2026-05-28 11:42:40.255794: 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])
2026-05-28 11:49:51.727086: 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.
2026-05-28 11:49:55.817810: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:50:02.439646: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-05-28 11:50:02.440628: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-05-28 11:50:02.468883: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:4a:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-05-28 11:50:02.468969: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:50:02.474668: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:50:02.474750: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:50:02.477064: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:50:02.478726: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:50:02.482619: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:50:02.484618: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:50:02.486288: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:50:02.486745: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:50:02.487162: 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.
2026-05-28 11:50:02.487346: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-05-28 11:50:02.487573: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:4a:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-05-28 11:50:02.487597: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:50:02.487612: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:50:02.487622: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:50:02.487633: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:50:02.487643: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:50:02.487653: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:50:02.487662: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:50:02.487673: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:50:02.487948: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:50:02.487969: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:50:02.930489: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-05-28 11:50:02.930572: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-05-28 11:50:02.930582: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-05-28 11:50:02.931291: 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:4a: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/80 [00:00<?, ?it/s]2026-05-28 11:50:04.690792: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2026-05-28 11:50:04.691339: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2500000000 Hz
2026-05-28 11:50:04.893332: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:50:05.503126: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:50:05.505209: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:50:07.420281: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-05-28 11:50:07.667817: 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/80 [00:32<43:06, 32.74s/it]
batch:   2%|▎         | 2/80 [00:32<17:42, 13.63s/it]
batch:   4%|▍         | 3/80 [00:33<09:38,  7.52s/it]
batch:   5%|▌         | 4/80 [00:33<05:53,  4.65s/it]
batch:   6%|▋         | 5/80 [00:33<03:49,  3.06s/it]
batch:   8%|▊         | 6/80 [00:33<02:35,  2.10s/it]
batch:   9%|▉         | 7/80 [00:34<01:48,  1.49s/it]
batch:  10%|█         | 8/80 [00:34<01:18,  1.09s/it]
batch:  11%|█▏        | 9/80 [00:34<00:58,  1.21it/s]
batch:  12%|█▎        | 10/80 [00:34<00:45,  1.55it/s]
batch:  14%|█▍        | 11/80 [00:35<00:35,  1.92it/s]
batch:  15%|█▌        | 12/80 [00:35<00:29,  2.30it/s]
batch:  16%|█▋        | 13/80 [00:35<00:25,  2.67it/s]
batch:  18%|█▊        | 14/80 [00:35<00:22,  3.00it/s]
batch:  19%|█▉        | 15/80 [00:36<00:19,  3.26it/s]
batch:  20%|██        | 16/80 [00:36<00:18,  3.48it/s]
batch:  21%|██▏       | 17/80 [00:36<00:17,  3.66it/s]
batch:  22%|██▎       | 18/80 [00:36<00:16,  3.78it/s]
batch:  24%|██▍       | 19/80 [00:37<00:15,  3.89it/s]
batch:  25%|██▌       | 20/80 [00:37<00:15,  3.95it/s]
batch:  26%|██▋       | 21/80 [00:37<00:14,  4.03it/s]
batch:  28%|██▊       | 22/80 [00:37<00:14,  4.04it/s]
batch:  29%|██▉       | 23/80 [00:38<00:14,  4.05it/s]
batch:  30%|███       | 24/80 [00:38<00:13,  4.08it/s]
batch:  31%|███▏      | 25/80 [00:38<00:13,  4.10it/s]
batch:  32%|███▎      | 26/80 [00:38<00:13,  4.13it/s]
batch:  34%|███▍      | 27/80 [00:39<00:13,  4.07it/s]
batch:  35%|███▌      | 28/80 [00:39<00:12,  4.06it/s]
batch:  36%|███▋      | 29/80 [00:39<00:12,  4.03it/s]
batch:  38%|███▊      | 30/80 [00:39<00:12,  4.08it/s]
batch:  39%|███▉      | 31/80 [00:40<00:12,  4.08it/s]
batch:  40%|████      | 32/80 [00:40<00:11,  4.09it/s]
batch:  41%|████▏     | 33/80 [00:40<00:11,  4.12it/s]
batch:  42%|████▎     | 34/80 [00:40<00:11,  4.11it/s]
batch:  44%|████▍     | 35/80 [00:40<00:10,  4.13it/s]
batch:  45%|████▌     | 36/80 [00:41<00:10,  4.11it/s]
batch:  46%|████▋     | 37/80 [00:41<00:10,  4.12it/s]
batch:  48%|████▊     | 38/80 [00:41<00:10,  4.10it/s]
batch:  49%|████▉     | 39/80 [00:41<00:09,  4.10it/s]
batch:  50%|█████     | 40/80 [00:42<00:09,  4.10it/s]
batch:  51%|█████▏    | 41/80 [00:42<00:09,  4.11it/s]
batch:  52%|█████▎    | 42/80 [00:42<00:09,  4.10it/s]
batch:  54%|█████▍    | 43/80 [00:42<00:08,  4.12it/s]
batch:  55%|█████▌    | 44/80 [00:43<00:08,  4.12it/s]
batch:  56%|█████▋    | 45/80 [00:43<00:08,  4.10it/s]
batch:  57%|█████▊    | 46/80 [00:43<00:08,  4.09it/s]
batch:  59%|█████▉    | 47/80 [00:43<00:08,  4.12it/s]
batch:  60%|██████    | 48/80 [00:44<00:07,  4.13it/s]
batch:  61%|██████▏   | 49/80 [00:44<00:07,  4.13it/s]
batch:  62%|██████▎   | 50/80 [00:44<00:07,  4.14it/s]
batch:  64%|██████▍   | 51/80 [00:44<00:06,  4.15it/s]
batch:  65%|██████▌   | 52/80 [00:45<00:06,  4.12it/s]
batch:  66%|██████▋   | 53/80 [00:45<00:06,  4.06it/s]
batch:  68%|██████▊   | 54/80 [00:45<00:06,  4.08it/s]
batch:  69%|██████▉   | 55/80 [00:45<00:06,  4.10it/s]
batch:  70%|███████   | 56/80 [00:46<00:05,  4.09it/s]
batch:  71%|███████▏  | 57/80 [00:46<00:05,  4.13it/s]
batch:  72%|███████▎  | 58/80 [00:46<00:05,  4.12it/s]
batch:  74%|███████▍  | 59/80 [00:46<00:05,  4.11it/s]
batch:  75%|███████▌  | 60/80 [00:47<00:04,  4.09it/s]
batch:  76%|███████▋  | 61/80 [00:47<00:04,  4.09it/s]
batch:  78%|███████▊  | 62/80 [00:47<00:04,  4.09it/s]
batch:  79%|███████▉  | 63/80 [00:47<00:04,  4.12it/s]
batch:  80%|████████  | 64/80 [00:48<00:03,  4.09it/s]
batch:  81%|████████▏ | 65/80 [00:48<00:03,  4.08it/s]
batch:  82%|████████▎ | 66/80 [00:48<00:04,  3.49it/s]
batch:  84%|████████▍ | 67/80 [00:48<00:03,  3.64it/s]
batch:  85%|████████▌ | 68/80 [00:49<00:03,  3.76it/s]
batch:  86%|████████▋ | 69/80 [00:49<00:02,  3.85it/s]
batch:  88%|████████▊ | 70/80 [00:49<00:02,  3.90it/s]
batch:  89%|████████▉ | 71/80 [00:49<00:02,  3.97it/s]
batch:  90%|█████████ | 72/80 [00:50<00:01,  4.01it/s]
batch:  91%|█████████▏| 73/80 [00:50<00:01,  4.03it/s]
batch:  92%|█████████▎| 74/80 [00:50<00:01,  4.07it/s]
batch:  94%|█████████▍| 75/80 [00:50<00:01,  4.08it/s]
batch:  95%|█████████▌| 76/80 [00:51<00:00,  4.07it/s]
batch:  96%|█████████▋| 77/80 [00:51<00:00,  4.08it/s]
batch:  98%|█████████▊| 78/80 [00:51<00:00,  4.08it/s]
batch:  99%|█████████▉| 79/80 [00:51<00:00,  4.09it/s]
batch: 100%|██████████| 80/80 [00:52<00:00,  4.60it/s]
batch: 100%|██████████| 80/80 [00:52<00:00,  1.54it/s]

  0%|          | 0/5091 [00:00<?, ?it/s]
  2%|▏         | 118/5091 [00:00<00:04, 1176.17it/s]
  5%|▍         | 236/5091 [00:00<00:04, 1177.70it/s]
  7%|▋         | 354/5091 [00:00<00:04, 1178.52it/s]
  9%|▉         | 472/5091 [00:00<00:03, 1170.89it/s]
 12%|█▏        | 590/5091 [00:00<00:03, 1171.07it/s]
 14%|█▍        | 708/5091 [00:00<00:03, 1169.10it/s]
 16%|█▌        | 825/5091 [00:00<00:03, 1165.26it/s]
 19%|█▊        | 944/5091 [00:00<00:03, 1167.68it/s]
 21%|██        | 1062/5091 [00:00<00:03, 1165.02it/s]
 23%|██▎       | 1179/5091 [00:01<00:03, 1166.30it/s]
 25%|██▌       | 1296/5091 [00:01<00:03, 1159.56it/s]
 28%|██▊       | 1413/5091 [00:01<00:03, 1156.05it/s]
 30%|███       | 1529/5091 [00:01<00:03, 1153.88it/s]
 32%|███▏      | 1645/5091 [00:01<00:02, 1149.33it/s]
 35%|███▍      | 1760/5091 [00:01<00:02, 1141.08it/s]
 37%|███▋      | 1875/5091 [00:01<00:02, 1139.13it/s]
 39%|███▉      | 1989/5091 [00:01<00:02, 1129.87it/s]
 41%|████▏     | 2102/5091 [00:01<00:02, 1126.97it/s]
 44%|████▎     | 2215/5091 [00:01<00:02, 1124.60it/s]
 46%|████▌     | 2329/5091 [00:02<00:02, 1123.57it/s]
 48%|████▊     | 2442/5091 [00:02<00:02, 1117.14it/s]
 50%|█████     | 2554/5091 [00:02<00:02, 1112.66it/s]
 52%|█████▏    | 2667/5091 [00:02<00:02, 1113.46it/s]
 55%|█████▍    | 2779/5091 [00:02<00:02, 1108.00it/s]
 57%|█████▋    | 2891/5091 [00:02<00:01, 1105.41it/s]
 59%|█████▉    | 3002/5091 [00:02<00:01, 1104.95it/s]
 61%|██████    | 3113/5091 [00:02<00:01, 1100.24it/s]
 63%|██████▎   | 3224/5091 [00:02<00:01, 1100.48it/s]
 66%|██████▌   | 3335/5091 [00:02<00:01, 1082.57it/s]
 68%|██████▊   | 3444/5091 [00:03<00:01, 1084.62it/s]
 70%|██████▉   | 3553/5091 [00:03<00:01, 1080.17it/s]
 72%|███████▏  | 3662/5091 [00:03<00:01, 1077.05it/s]
 74%|███████▍  | 3770/5091 [00:03<00:01, 1077.03it/s]
 76%|███████▌  | 3878/5091 [00:03<00:01, 1071.90it/s]
 78%|███████▊  | 3986/5091 [00:03<00:01, 1072.99it/s]
 80%|████████  | 4094/5091 [00:03<00:00, 1065.77it/s]
 83%|████████▎ | 4201/5091 [00:03<00:00, 1065.66it/s]
 85%|████████▍ | 4308/5091 [00:03<00:00, 1056.83it/s]
 87%|████████▋ | 4415/5091 [00:03<00:00, 1055.95it/s]
 89%|████████▉ | 4521/5091 [00:04<00:00, 1056.96it/s]
 91%|█████████ | 4627/5091 [00:04<00:00, 1051.00it/s]
 93%|█████████▎| 4733/5091 [00:04<00:00, 1047.14it/s]
 95%|█████████▌| 4838/5091 [00:04<00:00, 1045.39it/s]
 97%|█████████▋| 4943/5091 [00:04<00:00, 1045.26it/s]
 99%|█████████▉| 5048/5091 [00:04<00:00, 1039.75it/s]
100%|██████████| 5091/5091 [00:04<00:00, 1104.34it/s]

  0%|          | 0/5091 [00:00<?, ?it/s]
  2%|▏         | 118/5091 [00:00<00:04, 1175.92it/s]
  5%|▍         | 236/5091 [00:00<00:04, 1177.18it/s]
  7%|▋         | 354/5091 [00:00<00:04, 1177.20it/s]
  9%|▉         | 473/5091 [00:00<00:03, 1175.11it/s]
 12%|█▏        | 591/5091 [00:00<00:03, 1175.25it/s]
 14%|█▍        | 709/5091 [00:00<00:03, 1171.82it/s]
 16%|█▌        | 827/5091 [00:00<00:03, 1169.92it/s]
 19%|█▊        | 946/5091 [00:00<00:03, 1171.17it/s]
 21%|██        | 1064/5091 [00:00<00:03, 1173.32it/s]
 23%|██▎       | 1182/5091 [00:01<00:03, 1168.37it/s]
 26%|██▌       | 1299/5091 [00:01<00:03, 1161.01it/s]
 28%|██▊       | 1416/5091 [00:01<00:03, 1157.22it/s]
 30%|███       | 1532/5091 [00:01<00:03, 1154.85it/s]
 32%|███▏      | 1648/5091 [00:01<00:02, 1149.76it/s]
 35%|███▍      | 1763/5091 [00:01<00:02, 1141.84it/s]
 37%|███▋      | 1878/5091 [00:01<00:02, 1134.51it/s]
 39%|███▉      | 1992/5091 [00:01<00:02, 1126.02it/s]
 41%|████▏     | 2105/5091 [00:01<00:02, 1124.37it/s]
 44%|████▎     | 2218/5091 [00:01<00:02, 1123.55it/s]
 46%|████▌     | 2331/5091 [00:02<00:02, 1124.10it/s]
 48%|████▊     | 2444/5091 [00:02<00:02, 1095.16it/s]
 50%|█████     | 2557/5091 [00:02<00:02, 1099.23it/s]
 52%|█████▏    | 2670/5091 [00:02<00:02, 1103.04it/s]
 55%|█████▍    | 2781/5091 [00:03<00:06, 369.23it/s] 
 57%|█████▋    | 2891/5091 [00:03<00:04, 458.48it/s]
 59%|█████▉    | 3000/5091 [00:03<00:03, 552.73it/s]
 61%|██████    | 3110/5091 [00:03<00:03, 648.89it/s]
 63%|██████▎   | 3219/5091 [00:03<00:02, 737.41it/s]
 65%|██████▌   | 3328/5091 [00:03<00:02, 815.91it/s]
 67%|██████▋   | 3434/5091 [00:03<00:01, 874.52it/s]
 70%|██████▉   | 3543/5091 [00:03<00:01, 929.30it/s]
 72%|███████▏  | 3650/5091 [00:03<00:01, 964.25it/s]
 74%|███████▍  | 3758/5091 [00:04<00:01, 995.80it/s]
 76%|███████▌  | 3866/5091 [00:04<00:01, 1018.82it/s]
 78%|███████▊  | 3973/5091 [00:04<00:01, 1028.67it/s]
 80%|████████  | 4080/5091 [00:04<00:00, 1035.94it/s]
 82%|████████▏ | 4187/5091 [00:04<00:00, 1044.52it/s]
 84%|████████▍ | 4294/5091 [00:04<00:00, 1046.83it/s]
 86%|████████▋ | 4400/5091 [00:04<00:00, 1046.57it/s]
 89%|████████▊ | 4506/5091 [00:04<00:00, 1046.09it/s]
 91%|█████████ | 4612/5091 [00:04<00:00, 1048.68it/s]
 93%|█████████▎| 4718/5091 [00:04<00:00, 1038.88it/s]
 95%|█████████▍| 4823/5091 [00:05<00:00, 1036.56it/s]
 97%|█████████▋| 4927/5091 [00:05<00:00, 1035.03it/s]
 99%|█████████▉| 5031/5091 [00:05<00:00, 1030.51it/s]
100%|██████████| 5091/5091 [00:05<00:00, 961.30it/s] 
2026-05-28 11:51:22.817916: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:51:29.599272: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-05-28 11:51:29.600758: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-05-28 11:51:29.629142: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:4a:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-05-28 11:51:29.629245: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:51:29.633803: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:51:29.633898: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:51:29.636006: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:51:29.637296: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:51:29.640610: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:51:29.642379: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:51:29.643636: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:51:29.644079: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:51:29.644511: 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.
2026-05-28 11:51:29.644661: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-05-28 11:51:29.644895: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:4a:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-05-28 11:51:29.644926: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:51:29.644941: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:51:29.644955: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:51:29.644967: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:51:29.644980: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:51:29.644992: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:51:29.645023: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:51:29.645036: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:51:29.645326: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:51:29.645353: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:51:30.062739: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-05-28 11:51:30.062846: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-05-28 11:51:30.062857: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-05-28 11:51:30.063675: 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:4a:00.0, compute capability: 8.9)
2026-05-28 11:51:30.108929: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:196] None of the MLIR optimization passes are enabled (registered 0 passes)
2026-05-28 11:51:30.125174: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2500000000 Hz
2026-05-28 11:51:34.240099: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:51:34.760164: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:51:34.765142: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:51:36.358808: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-05-28 11:51:36.475148: 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.
2026-05-28 11:51:55.876847: 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)