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



Lmod Warning:
-------------------------------------------------------------------------------
The following dependent module(s) are not currently loaded: curl/8.17.0
(required by: ucsc-utils/v489), openssl/3.0.7 (required by: curl/8.4.0)
-------------------------------------------------------------------------------




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

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

2026-07-10 15:18:31.924479: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:20:01.750424: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 15:20:01.757869: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-07-10 15:20:02.024220: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-07-10 15:20:02.024297: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:20:06.211297: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:20:06.211388: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:20:06.217046: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 15:20:06.220637: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 15:20:06.225538: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 15:20:06.228024: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 15:20:06.230697: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:20:06.235616: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 15:20:06.235953: 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 FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2026-07-10 15:20:06.236025: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 15:20:06.238423: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-07-10 15:20:06.238450: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:20:06.238477: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:20:06.238492: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:20:06.238514: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 15:20:06.238528: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 15:20:06.238540: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 15:20:06.238552: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 15:20:06.238564: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:20:06.243262: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 15:20:06.243295: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:20:06.674265: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-07-10 15:20:06.674378: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-07-10 15:20:06.674393: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-07-10 15:20:06.679178: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 75642 MB memory) -> physical GPU (device: 0, name: NVIDIA A100-SXM4-80GB, pci bus id: 0000:84:00.0, compute capability: 8.0)
2026-07-10 15:20:16.143881: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2026-07-10 15:20:16.144327: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450010000 Hz
2026-07-10 15:20:18.910088: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:20:21.238055: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:20:21.252129: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:20:36.795039: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-07-10 15:20:36.914587: 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-07-10 15:21:26.259831: 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-07-10 15:39:23.581353: 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-07-10 15:39:27.950303: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:39:35.022952: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 15:39:35.023891: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-07-10 15:39:35.170677: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-07-10 15:39:35.170747: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:39:35.176077: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:39:35.176124: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:39:35.178360: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 15:39:35.180069: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 15:39:35.183421: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 15:39:35.185290: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 15:39:35.186924: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:39:35.193074: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 15:39:35.193340: 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 FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2026-07-10 15:39:35.193412: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 15:39:35.195820: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-07-10 15:39:35.195847: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:39:35.195865: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:39:35.195880: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:39:35.195894: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 15:39:35.195907: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 15:39:35.195921: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 15:39:35.195933: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 15:39:35.195946: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:39:35.200648: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 15:39:35.200682: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:39:35.651385: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-07-10 15:39:35.651505: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-07-10 15:39:35.651535: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-07-10 15:39:35.657198: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 75642 MB memory) -> physical GPU (device: 0, name: NVIDIA A100-SXM4-80GB, pci bus id: 0000:84:00.0, compute capability: 8.0)
/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/82 [00:00<?, ?it/s]2026-07-10 15:39:37.658317: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2026-07-10 15:39:37.658759: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450010000 Hz
2026-07-10 15:39:38.303007: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:39:39.575671: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:39:39.595523: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:39:43.505582: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-07-10 15:39:43.853754: 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/82 [00:38<52:11, 38.66s/it]
batch:   2%|▏         | 2/82 [00:38<21:25, 16.07s/it]
batch:   4%|▎         | 3/82 [00:39<11:39,  8.85s/it]
batch:   5%|▍         | 4/82 [00:39<07:05,  5.46s/it]
batch:   6%|▌         | 5/82 [00:39<04:36,  3.58s/it]
batch:   7%|▋         | 6/82 [00:39<03:06,  2.45s/it]
batch:   9%|▊         | 7/82 [00:40<02:10,  1.74s/it]
batch:  10%|▉         | 8/82 [00:40<01:33,  1.26s/it]
batch:  11%|█         | 9/82 [00:40<01:09,  1.05it/s]
batch:  12%|█▏        | 10/82 [00:41<00:53,  1.35it/s]
batch:  13%|█▎        | 11/82 [00:41<00:42,  1.69it/s]
batch:  15%|█▍        | 12/82 [00:41<00:34,  2.04it/s]
batch:  16%|█▌        | 13/82 [00:41<00:28,  2.38it/s]
batch:  17%|█▋        | 14/82 [00:42<00:25,  2.69it/s]
batch:  18%|█▊        | 15/82 [00:42<00:22,  2.97it/s]
batch:  20%|█▉        | 16/82 [00:42<00:20,  3.20it/s]
batch:  21%|██        | 17/82 [00:42<00:19,  3.36it/s]
batch:  22%|██▏       | 18/82 [00:43<00:18,  3.51it/s]
batch:  23%|██▎       | 19/82 [00:43<00:17,  3.61it/s]
batch:  24%|██▍       | 20/82 [00:43<00:16,  3.69it/s]
batch:  26%|██▌       | 21/82 [00:43<00:16,  3.74it/s]
batch:  27%|██▋       | 22/82 [00:44<00:15,  3.79it/s]
batch:  28%|██▊       | 23/82 [00:44<00:15,  3.82it/s]
batch:  29%|██▉       | 24/82 [00:44<00:16,  3.43it/s]
batch:  30%|███       | 25/82 [00:44<00:16,  3.55it/s]
batch:  32%|███▏      | 26/82 [00:45<00:15,  3.63it/s]
batch:  33%|███▎      | 27/82 [00:45<00:14,  3.71it/s]
batch:  34%|███▍      | 28/82 [00:45<00:14,  3.76it/s]
batch:  35%|███▌      | 29/82 [00:46<00:14,  3.79it/s]
batch:  37%|███▋      | 30/82 [00:46<00:13,  3.82it/s]
batch:  38%|███▊      | 31/82 [00:46<00:13,  3.84it/s]
batch:  39%|███▉      | 32/82 [00:46<00:12,  3.85it/s]
batch:  40%|████      | 33/82 [00:47<00:12,  3.87it/s]
batch:  41%|████▏     | 34/82 [00:47<00:12,  3.87it/s]
batch:  43%|████▎     | 35/82 [00:47<00:12,  3.88it/s]
batch:  44%|████▍     | 36/82 [00:47<00:11,  3.87it/s]
batch:  45%|████▌     | 37/82 [00:48<00:11,  3.87it/s]
batch:  46%|████▋     | 38/82 [00:48<00:11,  3.88it/s]
batch:  48%|████▊     | 39/82 [00:48<00:11,  3.84it/s]
batch:  49%|████▉     | 40/82 [00:48<00:10,  3.85it/s]
batch:  50%|█████     | 41/82 [00:49<00:11,  3.69it/s]
batch:  51%|█████     | 42/82 [00:49<00:10,  3.75it/s]
batch:  52%|█████▏    | 43/82 [00:49<00:10,  3.79it/s]
batch:  54%|█████▎    | 44/82 [00:49<00:09,  3.83it/s]
batch:  55%|█████▍    | 45/82 [00:50<00:09,  3.85it/s]
batch:  56%|█████▌    | 46/82 [00:50<00:09,  3.86it/s]
batch:  57%|█████▋    | 47/82 [00:50<00:09,  3.87it/s]
batch:  59%|█████▊    | 48/82 [00:50<00:08,  3.86it/s]
batch:  60%|█████▉    | 49/82 [00:51<00:08,  3.87it/s]
batch:  61%|██████    | 50/82 [00:51<00:08,  3.87it/s]
batch:  62%|██████▏   | 51/82 [00:51<00:08,  3.87it/s]
batch:  63%|██████▎   | 52/82 [00:51<00:07,  3.87it/s]
batch:  65%|██████▍   | 53/82 [00:52<00:07,  3.88it/s]
batch:  66%|██████▌   | 54/82 [00:52<00:07,  3.88it/s]
batch:  67%|██████▋   | 55/82 [00:52<00:06,  3.87it/s]
batch:  68%|██████▊   | 56/82 [00:53<00:06,  3.87it/s]
batch:  70%|██████▉   | 57/82 [00:53<00:06,  3.88it/s]
batch:  71%|███████   | 58/82 [00:53<00:06,  3.87it/s]
batch:  72%|███████▏  | 59/82 [00:53<00:05,  3.88it/s]
batch:  73%|███████▎  | 60/82 [00:54<00:05,  3.88it/s]
batch:  74%|███████▍  | 61/82 [00:54<00:05,  3.87it/s]
batch:  76%|███████▌  | 62/82 [00:54<00:05,  3.88it/s]
batch:  77%|███████▋  | 63/82 [00:54<00:04,  3.87it/s]
batch:  78%|███████▊  | 64/82 [00:55<00:04,  3.86it/s]
batch:  79%|███████▉  | 65/82 [00:55<00:04,  3.87it/s]
batch:  80%|████████  | 66/82 [00:55<00:04,  3.88it/s]
batch:  82%|████████▏ | 67/82 [00:55<00:03,  3.87it/s]
batch:  83%|████████▎ | 68/82 [00:56<00:03,  3.86it/s]
batch:  84%|████████▍ | 69/82 [00:56<00:03,  3.86it/s]
batch:  85%|████████▌ | 70/82 [00:56<00:03,  3.85it/s]
batch:  87%|████████▋ | 71/82 [00:56<00:02,  3.82it/s]
batch:  88%|████████▊ | 72/82 [00:57<00:02,  3.84it/s]
batch:  89%|████████▉ | 73/82 [00:57<00:02,  3.85it/s]
batch:  90%|█████████ | 74/82 [00:57<00:02,  3.87it/s]
batch:  91%|█████████▏| 75/82 [00:57<00:01,  3.87it/s]
batch:  93%|█████████▎| 76/82 [00:58<00:01,  3.88it/s]
batch:  94%|█████████▍| 77/82 [00:58<00:01,  3.88it/s]
batch:  95%|█████████▌| 78/82 [00:58<00:01,  3.88it/s]
batch:  96%|█████████▋| 79/82 [00:58<00:00,  3.88it/s]
batch:  98%|█████████▊| 80/82 [00:59<00:00,  3.88it/s]
batch:  99%|█████████▉| 81/82 [00:59<00:00,  4.34it/s]
batch: 100%|██████████| 82/82 [00:59<00:00,  1.38it/s]

  0%|          | 0/5156 [00:00<?, ?it/s]
  2%|▏         | 112/5156 [00:00<00:04, 1118.64it/s]
  4%|▍         | 226/5156 [00:00<00:04, 1129.77it/s]
  7%|▋         | 339/5156 [00:00<00:04, 1122.59it/s]
  9%|▉         | 452/5156 [00:00<00:04, 1124.45it/s]
 11%|█         | 565/5156 [00:00<00:04, 1123.29it/s]
 13%|█▎        | 678/5156 [00:00<00:03, 1121.25it/s]
 15%|█▌        | 791/5156 [00:00<00:03, 1114.09it/s]
 18%|█▊        | 903/5156 [00:00<00:03, 1111.47it/s]
 20%|█▉        | 1015/5156 [00:00<00:03, 1110.57it/s]
 22%|██▏       | 1127/5156 [00:01<00:03, 1079.65it/s]
 24%|██▍       | 1239/5156 [00:01<00:03, 1086.82it/s]
 26%|██▌       | 1349/5156 [00:01<00:03, 1090.72it/s]
 28%|██▊       | 1459/5156 [00:01<00:03, 1091.09it/s]
 30%|███       | 1569/5156 [00:01<00:03, 1091.40it/s]
 33%|███▎      | 1679/5156 [00:01<00:03, 1091.37it/s]
 35%|███▍      | 1789/5156 [00:01<00:03, 1091.71it/s]
 37%|███▋      | 1899/5156 [00:01<00:02, 1086.17it/s]
 39%|███▉      | 2008/5156 [00:01<00:02, 1083.40it/s]
 41%|████      | 2118/5156 [00:01<00:02, 1082.48it/s]
 43%|████▎     | 2227/5156 [00:02<00:02, 1083.91it/s]
 45%|████▌     | 2336/5156 [00:02<00:02, 1077.66it/s]
 47%|████▋     | 2444/5156 [00:02<00:02, 1071.60it/s]
 49%|████▉     | 2552/5156 [00:02<00:02, 1069.42it/s]
 52%|█████▏    | 2659/5156 [00:02<00:02, 1068.48it/s]
 54%|█████▎    | 2766/5156 [00:02<00:02, 1066.08it/s]
 56%|█████▌    | 2873/5156 [00:02<00:02, 1063.35it/s]
 58%|█████▊    | 2980/5156 [00:02<00:02, 1060.63it/s]
 60%|█████▉    | 3087/5156 [00:02<00:01, 1057.44it/s]
 62%|██████▏   | 3193/5156 [00:02<00:01, 1055.33it/s]
 64%|██████▍   | 3299/5156 [00:03<00:01, 1050.38it/s]
 66%|██████▌   | 3405/5156 [00:03<00:01, 1046.62it/s]
 68%|██████▊   | 3510/5156 [00:03<00:01, 1046.43it/s]
 70%|███████   | 3615/5156 [00:03<00:01, 1038.38it/s]
 72%|███████▏  | 3719/5156 [00:03<00:01, 1028.60it/s]
 74%|███████▍  | 3824/5156 [00:03<00:01, 1029.80it/s]
 76%|███████▌  | 3928/5156 [00:03<00:01, 1028.83it/s]
 78%|███████▊  | 4032/5156 [00:03<00:01, 1031.81it/s]
 80%|████████  | 4136/5156 [00:03<00:00, 1023.47it/s]
 82%|████████▏ | 4240/5156 [00:03<00:00, 1024.75it/s]
 84%|████████▍ | 4344/5156 [00:04<00:00, 1025.32it/s]
 86%|████████▌ | 4447/5156 [00:04<00:00, 1022.08it/s]
 88%|████████▊ | 4551/5156 [00:04<00:00, 1023.35it/s]
 90%|█████████ | 4655/5156 [00:04<00:00, 1024.64it/s]
 92%|█████████▏| 4758/5156 [00:04<00:00, 1020.33it/s]
 94%|█████████▍| 4861/5156 [00:04<00:00, 1021.07it/s]
 96%|█████████▋| 4964/5156 [00:04<00:00, 1018.27it/s]
 98%|█████████▊| 5066/5156 [00:04<00:00, 1013.02it/s]
100%|██████████| 5156/5156 [00:04<00:00, 1060.49it/s]

  0%|          | 0/5156 [00:00<?, ?it/s]
  2%|▏         | 114/5156 [00:00<00:04, 1137.61it/s]
  4%|▍         | 228/5156 [00:00<00:04, 1127.41it/s]
  7%|▋         | 342/5156 [00:00<00:04, 1126.71it/s]
  9%|▉         | 457/5156 [00:00<00:04, 1127.45it/s]
 11%|█         | 570/5156 [00:00<00:04, 1127.14it/s]
 13%|█▎        | 683/5156 [00:00<00:03, 1126.14it/s]
 15%|█▌        | 796/5156 [00:00<00:03, 1125.54it/s]
 18%|█▊        | 909/5156 [00:00<00:03, 1117.04it/s]
 20%|█▉        | 1021/5156 [00:00<00:03, 1115.67it/s]
 22%|██▏       | 1133/5156 [00:01<00:03, 1115.58it/s]
 24%|██▍       | 1245/5156 [00:01<00:03, 1113.50it/s]
 26%|██▋       | 1357/5156 [00:01<00:03, 1110.50it/s]
 28%|██▊       | 1469/5156 [00:01<00:03, 1110.41it/s]
 31%|███       | 1581/5156 [00:01<00:03, 1105.19it/s]
 33%|███▎      | 1692/5156 [00:01<00:03, 1099.67it/s]
 35%|███▍      | 1802/5156 [00:01<00:03, 1098.97it/s]
 37%|███▋      | 1912/5156 [00:01<00:02, 1097.01it/s]
 39%|███▉      | 2022/5156 [00:01<00:02, 1085.43it/s]
 41%|████▏     | 2131/5156 [00:01<00:02, 1084.36it/s]
 43%|████▎     | 2240/5156 [00:02<00:02, 1082.31it/s]
 46%|████▌     | 2349/5156 [00:02<00:02, 1076.63it/s]
 48%|████▊     | 2457/5156 [00:02<00:02, 1030.01it/s]
 50%|████▉     | 2563/5156 [00:02<00:02, 1036.40it/s]
 52%|█████▏    | 2670/5156 [00:02<00:02, 1045.01it/s]
 54%|█████▍    | 2777/5156 [00:02<00:02, 1051.98it/s]
 56%|█████▌    | 2884/5156 [00:02<00:02, 1054.05it/s]
 58%|█████▊    | 2991/5156 [00:02<00:02, 1055.85it/s]
 60%|██████    | 3097/5156 [00:02<00:01, 1056.64it/s]
 62%|██████▏   | 3203/5156 [00:02<00:01, 1049.25it/s]
 64%|██████▍   | 3308/5156 [00:03<00:01, 1049.21it/s]
 66%|██████▌   | 3413/5156 [00:03<00:01, 1044.56it/s]
 68%|██████▊   | 3518/5156 [00:03<00:01, 1038.41it/s]
 70%|███████   | 3622/5156 [00:03<00:01, 1037.80it/s]
 72%|███████▏  | 3726/5156 [00:03<00:01, 1035.47it/s]
 74%|███████▍  | 3830/5156 [00:03<00:01, 1035.44it/s]
 76%|███████▋  | 3934/5156 [00:03<00:01, 1032.99it/s]
 78%|███████▊  | 4038/5156 [00:03<00:01, 1030.02it/s]
 80%|████████  | 4142/5156 [00:03<00:00, 1028.89it/s]
 82%|████████▏ | 4245/5156 [00:03<00:00, 1027.25it/s]
 84%|████████▍ | 4349/5156 [00:04<00:00, 1027.73it/s]
 86%|████████▋ | 4453/5156 [00:04<00:00, 1025.91it/s]
 88%|████████▊ | 4557/5156 [00:04<00:00, 1026.22it/s]
 90%|█████████ | 4660/5156 [00:04<00:00, 1024.05it/s]
 92%|█████████▏| 4763/5156 [00:04<00:00, 1025.57it/s]
 94%|█████████▍| 4866/5156 [00:04<00:00, 1019.30it/s]
 96%|█████████▋| 4968/5156 [00:04<00:00, 1016.61it/s]
 98%|█████████▊| 5070/5156 [00:04<00:00, 1017.05it/s]
100%|██████████| 5156/5156 [00:04<00:00, 1061.96it/s]
2026-07-10 15:40:59.746798: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:41:06.820708: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 15:41:06.821699: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-07-10 15:41:06.967691: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-07-10 15:41:06.967808: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:41:06.972509: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:41:06.972561: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:41:06.974721: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 15:41:06.976160: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 15:41:06.979282: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 15:41:06.981024: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 15:41:06.982437: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:41:06.989396: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 15:41:06.989731: 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 FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2026-07-10 15:41:06.989802: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 15:41:06.993797: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-07-10 15:41:06.993847: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:41:06.993870: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:41:06.993889: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:41:06.993907: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 15:41:06.993924: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 15:41:06.993941: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 15:41:06.993958: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 15:41:06.993976: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:41:06.998698: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 15:41:06.998740: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:41:07.448108: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-07-10 15:41:07.448232: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-07-10 15:41:07.448245: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-07-10 15:41:07.453069: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 75642 MB memory) -> physical GPU (device: 0, name: NVIDIA A100-SXM4-80GB, pci bus id: 0000:84:00.0, compute capability: 8.0)
2026-07-10 15:41:07.494703: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:196] None of the MLIR optimization passes are enabled (registered 0 passes)
2026-07-10 15:41:07.505331: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450010000 Hz
2026-07-10 15:41:12.070083: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:41:12.464075: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:41:12.470097: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:41:13.724090: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-07-10 15:41:13.837217: 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-07-10 15:41:33.433154: 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)