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 14:55:46.028970: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 14:57:16.254937: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 14:57:16.273014: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-07-10 14:57:16.545227: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:c3: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 14:57:16.545373: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 14:57:17.107027: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 14:57:17.107219: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 14:57:17.583204: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 14:57:18.170245: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 14:57:18.737143: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 14:57:18.952604: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 14:57:19.121497: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 14:57:19.130056: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 14:57:19.130475: 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 14:57:19.130573: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 14:57:19.133007: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:c3: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 14:57:19.133044: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 14:57:19.133068: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 14:57:19.133083: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 14:57:19.133112: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 14:57:19.133128: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 14:57:19.133142: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 14:57:19.133155: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 14:57:19.133169: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 14:57:19.137885: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 14:57:19.137925: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 14:57:21.728728: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-07-10 14:57:21.728829: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-07-10 14:57:21.728846: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-07-10 14:57:21.732767: 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:c3:00.0, compute capability: 8.0)
2026-07-10 14:57:34.877023: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2026-07-10 14:57:34.877562: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2449915000 Hz
2026-07-10 14:57:38.123395: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 14:57:39.908697: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 14:57:39.913876: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 14:57:55.046271: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-07-10 14:57:55.354592: 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 14:58:31.617536: 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:16:52.957300: 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:16:59.537726: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:17:42.627617: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 15:17:42.628872: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-07-10 15:17:42.928099: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:c3: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:17:42.928231: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:17:42.937961: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:17:42.938072: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:17:42.942899: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 15:17:42.946500: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 15:17:42.951410: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 15:17:42.954477: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 15:17:42.957770: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:17:42.962259: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 15:17:42.962603: 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:17:42.962701: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 15:17:42.964890: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:c3: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:17:42.964928: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:17:42.964952: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:17:42.964970: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:17:42.964986: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 15:17:42.965002: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 15:17:42.965017: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 15:17:42.965031: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 15:17:42.965047: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:17:42.971763: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 15:17:42.971816: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:17:43.471417: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-07-10 15:17:43.471535: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-07-10 15:17:43.471570: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-07-10 15:17:43.475691: 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:c3: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:17:45.866145: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2026-07-10 15:17:45.866766: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2449915000 Hz
2026-07-10 15:17:46.751430: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:17:48.367738: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:17:48.387664: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:17:53.513821: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-07-10 15:17:53.972975: 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 [01:12<1:37:21, 72.12s/it]
batch:   2%|▏         | 2/82 [01:12<39:50, 29.88s/it]  
batch:   4%|▎         | 3/82 [01:12<21:33, 16.37s/it]
batch:   5%|▍         | 4/82 [01:13<13:02, 10.03s/it]
batch:   6%|▌         | 5/82 [01:13<08:22,  6.52s/it]
batch:   7%|▋         | 6/82 [01:13<05:35,  4.41s/it]
batch:   9%|▊         | 7/82 [01:13<03:50,  3.07s/it]
batch:  10%|▉         | 8/82 [01:14<02:41,  2.19s/it]
batch:  11%|█         | 9/82 [01:14<01:56,  1.60s/it]
batch:  12%|█▏        | 10/82 [01:14<01:26,  1.20s/it]
batch:  13%|█▎        | 11/82 [01:15<01:05,  1.08it/s]
batch:  15%|█▍        | 12/82 [01:15<00:51,  1.36it/s]
batch:  16%|█▌        | 13/82 [01:15<00:41,  1.66it/s]
batch:  17%|█▋        | 14/82 [01:16<00:34,  1.95it/s]
batch:  18%|█▊        | 15/82 [01:16<00:30,  2.23it/s]
batch:  20%|█▉        | 16/82 [01:16<00:26,  2.47it/s]
batch:  21%|██        | 17/82 [01:16<00:24,  2.67it/s]
batch:  22%|██▏       | 18/82 [01:17<00:22,  2.83it/s]
batch:  23%|██▎       | 19/82 [01:17<00:21,  2.95it/s]
batch:  24%|██▍       | 20/82 [01:17<00:20,  3.06it/s]
batch:  26%|██▌       | 21/82 [01:18<00:19,  3.14it/s]
batch:  27%|██▋       | 22/82 [01:18<00:18,  3.19it/s]
batch:  28%|██▊       | 23/82 [01:18<00:18,  3.22it/s]
batch:  29%|██▉       | 24/82 [01:19<00:20,  2.88it/s]
batch:  30%|███       | 25/82 [01:19<00:19,  2.99it/s]
batch:  32%|███▏      | 26/82 [01:19<00:18,  3.08it/s]
batch:  33%|███▎      | 27/82 [01:20<00:17,  3.15it/s]
batch:  34%|███▍      | 28/82 [01:20<00:16,  3.20it/s]
batch:  35%|███▌      | 29/82 [01:20<00:16,  3.24it/s]
batch:  37%|███▋      | 30/82 [01:21<00:15,  3.26it/s]
batch:  38%|███▊      | 31/82 [01:21<00:15,  3.27it/s]
batch:  39%|███▉      | 32/82 [01:21<00:15,  3.29it/s]
batch:  40%|████      | 33/82 [01:21<00:14,  3.30it/s]
batch:  41%|████▏     | 34/82 [01:22<00:14,  3.29it/s]
batch:  43%|████▎     | 35/82 [01:22<00:14,  3.31it/s]
batch:  44%|████▍     | 36/82 [01:22<00:13,  3.31it/s]
batch:  45%|████▌     | 37/82 [01:23<00:13,  3.31it/s]
batch:  46%|████▋     | 38/82 [01:23<00:13,  3.33it/s]
batch:  48%|████▊     | 39/82 [01:23<00:12,  3.32it/s]
batch:  49%|████▉     | 40/82 [01:24<00:12,  3.31it/s]
batch:  50%|█████     | 41/82 [01:24<00:12,  3.33it/s]
batch:  51%|█████     | 42/82 [01:24<00:12,  3.32it/s]
batch:  52%|█████▏    | 43/82 [01:24<00:11,  3.32it/s]
batch:  54%|█████▎    | 44/82 [01:25<00:11,  3.33it/s]
batch:  55%|█████▍    | 45/82 [01:25<00:11,  3.33it/s]
batch:  56%|█████▌    | 46/82 [01:25<00:10,  3.33it/s]
batch:  57%|█████▋    | 47/82 [01:26<00:10,  3.33it/s]
batch:  59%|█████▊    | 48/82 [01:26<00:10,  3.32it/s]
batch:  60%|█████▉    | 49/82 [01:26<00:09,  3.32it/s]
batch:  61%|██████    | 50/82 [01:27<00:09,  3.33it/s]
batch:  62%|██████▏   | 51/82 [01:27<00:09,  3.31it/s]
batch:  63%|██████▎   | 52/82 [01:27<00:09,  3.31it/s]
batch:  65%|██████▍   | 53/82 [01:27<00:08,  3.31it/s]
batch:  66%|██████▌   | 54/82 [01:28<00:08,  3.31it/s]
batch:  67%|██████▋   | 55/82 [01:28<00:08,  3.31it/s]
batch:  68%|██████▊   | 56/82 [01:28<00:07,  3.30it/s]
batch:  70%|██████▉   | 57/82 [01:29<00:07,  3.30it/s]
batch:  71%|███████   | 58/82 [01:29<00:07,  3.30it/s]
batch:  72%|███████▏  | 59/82 [01:29<00:06,  3.31it/s]
batch:  73%|███████▎  | 60/82 [01:30<00:06,  3.30it/s]
batch:  74%|███████▍  | 61/82 [01:30<00:06,  3.30it/s]
batch:  76%|███████▌  | 62/82 [01:30<00:06,  3.29it/s]
batch:  77%|███████▋  | 63/82 [01:31<00:05,  3.28it/s]
batch:  78%|███████▊  | 64/82 [01:31<00:05,  3.28it/s]
batch:  79%|███████▉  | 65/82 [01:31<00:05,  3.30it/s]
batch:  80%|████████  | 66/82 [01:31<00:04,  3.30it/s]
batch:  82%|████████▏ | 67/82 [01:32<00:04,  3.30it/s]
batch:  83%|████████▎ | 68/82 [01:32<00:04,  3.29it/s]
batch:  84%|████████▍ | 69/82 [01:32<00:03,  3.29it/s]
batch:  85%|████████▌ | 70/82 [01:33<00:03,  3.29it/s]
batch:  87%|████████▋ | 71/82 [01:33<00:03,  3.29it/s]
batch:  88%|████████▊ | 72/82 [01:33<00:03,  3.31it/s]
batch:  89%|████████▉ | 73/82 [01:34<00:02,  3.29it/s]
batch:  90%|█████████ | 74/82 [01:34<00:02,  3.30it/s]
batch:  91%|█████████▏| 75/82 [01:34<00:02,  3.30it/s]
batch:  93%|█████████▎| 76/82 [01:34<00:01,  3.27it/s]
batch:  94%|█████████▍| 77/82 [01:35<00:01,  3.27it/s]
batch:  95%|█████████▌| 78/82 [01:35<00:01,  3.28it/s]
batch:  96%|█████████▋| 79/82 [01:35<00:00,  3.29it/s]
batch:  98%|█████████▊| 80/82 [01:36<00:00,  3.27it/s]
batch:  99%|█████████▉| 81/82 [01:36<00:00,  3.68it/s]
batch: 100%|██████████| 82/82 [01:36<00:00,  1.18s/it]

  0%|          | 0/5156 [00:00<?, ?it/s]
  2%|▏         | 99/5156 [00:00<00:05, 985.46it/s]
  4%|▍         | 199/5156 [00:00<00:04, 992.20it/s]
  6%|▌         | 299/5156 [00:00<00:04, 988.18it/s]
  8%|▊         | 399/5156 [00:00<00:04, 984.30it/s]
 10%|▉         | 499/5156 [00:00<00:04, 986.33it/s]
 12%|█▏        | 598/5156 [00:00<00:04, 984.22it/s]
 14%|█▎        | 697/5156 [00:00<00:04, 982.17it/s]
 15%|█▌        | 796/5156 [00:00<00:04, 979.73it/s]
 17%|█▋        | 894/5156 [00:00<00:04, 975.94it/s]
 19%|█▉        | 992/5156 [00:01<00:04, 972.17it/s]
 21%|██        | 1090/5156 [00:01<00:04, 971.70it/s]
 23%|██▎       | 1188/5156 [00:01<00:04, 968.63it/s]
 25%|██▍       | 1286/5156 [00:01<00:04, 967.13it/s]
 27%|██▋       | 1383/5156 [00:01<00:03, 967.34it/s]
 29%|██▊       | 1480/5156 [00:01<00:03, 965.11it/s]
 31%|███       | 1577/5156 [00:01<00:03, 958.94it/s]
 32%|███▏      | 1673/5156 [00:01<00:03, 958.03it/s]
 34%|███▍      | 1769/5156 [00:01<00:03, 957.45it/s]
 36%|███▌      | 1865/5156 [00:01<00:03, 954.96it/s]
 38%|███▊      | 1961/5156 [00:02<00:03, 947.68it/s]
 40%|███▉      | 2056/5156 [00:02<00:03, 943.94it/s]
 42%|████▏     | 2151/5156 [00:02<00:03, 943.76it/s]
 44%|████▎     | 2246/5156 [00:02<00:03, 940.90it/s]
 45%|████▌     | 2341/5156 [00:02<00:03, 936.36it/s]
 47%|████▋     | 2435/5156 [00:02<00:02, 935.99it/s]
 49%|████▉     | 2529/5156 [00:02<00:02, 933.54it/s]
 51%|█████     | 2623/5156 [00:02<00:02, 929.43it/s]
 53%|█████▎    | 2716/5156 [00:02<00:02, 928.64it/s]
 54%|█████▍    | 2809/5156 [00:02<00:02, 928.42it/s]
 56%|█████▋    | 2904/5156 [00:03<00:02, 929.50it/s]
 58%|█████▊    | 2997/5156 [00:03<00:02, 927.15it/s]
 60%|█████▉    | 3090/5156 [00:03<00:02, 922.50it/s]
 62%|██████▏   | 3183/5156 [00:03<00:02, 917.91it/s]
 64%|██████▎   | 3276/5156 [00:03<00:02, 916.52it/s]
 65%|██████▌   | 3369/5156 [00:03<00:01, 915.46it/s]
 67%|██████▋   | 3461/5156 [00:03<00:01, 913.62it/s]
 69%|██████▉   | 3553/5156 [00:03<00:01, 913.58it/s]
 71%|███████   | 3645/5156 [00:03<00:01, 907.46it/s]
 72%|███████▏  | 3736/5156 [00:03<00:01, 905.82it/s]
 74%|███████▍  | 3827/5156 [00:04<00:01, 906.90it/s]
 76%|███████▌  | 3918/5156 [00:04<00:01, 900.16it/s]
 78%|███████▊  | 4009/5156 [00:04<00:01, 898.96it/s]
 79%|███████▉  | 4099/5156 [00:04<00:01, 896.65it/s]
 81%|████████▏ | 4190/5156 [00:04<00:01, 899.88it/s]
 83%|████████▎ | 4280/5156 [00:04<00:00, 895.63it/s]
 85%|████████▍ | 4371/5156 [00:04<00:00, 896.78it/s]
 87%|████████▋ | 4461/5156 [00:04<00:00, 894.14it/s]
 88%|████████▊ | 4551/5156 [00:04<00:00, 894.33it/s]
 90%|█████████ | 4641/5156 [00:04<00:00, 894.13it/s]
 92%|█████████▏| 4731/5156 [00:05<00:00, 893.47it/s]
 94%|█████████▎| 4822/5156 [00:05<00:00, 893.73it/s]
 95%|█████████▌| 4912/5156 [00:05<00:00, 893.09it/s]
 97%|█████████▋| 5002/5156 [00:05<00:00, 889.63it/s]
 99%|█████████▊| 5091/5156 [00:05<00:00, 886.89it/s]
100%|██████████| 5156/5156 [00:05<00:00, 929.58it/s]

  0%|          | 0/5156 [00:00<?, ?it/s]
  2%|▏         | 99/5156 [00:00<00:05, 989.15it/s]
  4%|▍         | 199/5156 [00:00<00:04, 994.37it/s]
  6%|▌         | 299/5156 [00:00<00:04, 989.40it/s]
  8%|▊         | 399/5156 [00:00<00:04, 985.02it/s]
 10%|▉         | 499/5156 [00:00<00:04, 986.98it/s]
 12%|█▏        | 598/5156 [00:00<00:04, 985.59it/s]
 14%|█▎        | 697/5156 [00:00<00:04, 984.20it/s]
 15%|█▌        | 796/5156 [00:00<00:04, 977.60it/s]
 17%|█▋        | 894/5156 [00:00<00:04, 975.29it/s]
 19%|█▉        | 992/5156 [00:01<00:04, 974.24it/s]
 21%|██        | 1090/5156 [00:01<00:04, 974.15it/s]
 23%|██▎       | 1188/5156 [00:01<00:04, 972.30it/s]
 25%|██▍       | 1286/5156 [00:01<00:03, 968.71it/s]
 27%|██▋       | 1384/5156 [00:01<00:03, 967.56it/s]
 29%|██▊       | 1481/5156 [00:01<00:03, 964.36it/s]
 31%|███       | 1578/5156 [00:01<00:03, 963.39it/s]
 32%|███▏      | 1675/5156 [00:01<00:03, 962.54it/s]
 34%|███▍      | 1772/5156 [00:01<00:03, 957.88it/s]
 36%|███▌      | 1868/5156 [00:01<00:03, 956.65it/s]
 38%|███▊      | 1964/5156 [00:02<00:03, 955.51it/s]
 40%|███▉      | 2060/5156 [00:02<00:03, 947.08it/s]
 42%|████▏     | 2155/5156 [00:02<00:03, 947.07it/s]
 44%|████▎     | 2250/5156 [00:02<00:03, 943.99it/s]
 45%|████▌     | 2345/5156 [00:02<00:02, 944.76it/s]
 47%|████▋     | 2440/5156 [00:02<00:02, 940.01it/s]
 49%|████▉     | 2535/5156 [00:02<00:02, 933.56it/s]
 51%|█████     | 2629/5156 [00:02<00:02, 934.16it/s]
 53%|█████▎    | 2723/5156 [00:02<00:02, 934.47it/s]
 55%|█████▍    | 2817/5156 [00:02<00:02, 925.53it/s]
 56%|█████▋    | 2912/5156 [00:03<00:02, 927.75it/s]
 58%|█████▊    | 3005/5156 [00:03<00:02, 926.51it/s]
 60%|██████    | 3098/5156 [00:03<00:02, 922.63it/s]
 62%|██████▏   | 3191/5156 [00:03<00:02, 917.59it/s]
 64%|██████▎   | 3284/5156 [00:03<00:02, 916.27it/s]
 65%|██████▌   | 3377/5156 [00:03<00:01, 915.27it/s]
 67%|██████▋   | 3470/5156 [00:03<00:01, 915.16it/s]
 69%|██████▉   | 3562/5156 [00:03<00:01, 914.84it/s]
 71%|███████   | 3654/5156 [00:03<00:01, 910.70it/s]
 73%|███████▎  | 3746/5156 [00:03<00:01, 911.19it/s]
 74%|███████▍  | 3838/5156 [00:04<00:01, 907.81it/s]
 76%|███████▌  | 3929/5156 [00:04<00:01, 907.47it/s]
 78%|███████▊  | 4020/5156 [00:04<00:01, 905.11it/s]
 80%|███████▉  | 4111/5156 [00:04<00:01, 903.68it/s]
 81%|████████▏ | 4202/5156 [00:04<00:01, 900.33it/s]
 83%|████████▎ | 4293/5156 [00:04<00:00, 902.86it/s]
 85%|████████▌ | 4384/5156 [00:04<00:00, 897.49it/s]
 87%|████████▋ | 4475/5156 [00:04<00:00, 897.15it/s]
 89%|████████▊ | 4565/5156 [00:04<00:00, 897.62it/s]
 90%|█████████ | 4655/5156 [00:04<00:00, 896.01it/s]
 92%|█████████▏| 4745/5156 [00:05<00:00, 889.81it/s]
 94%|█████████▍| 4835/5156 [00:05<00:00, 888.10it/s]
 96%|█████████▌| 4924/5156 [00:05<00:00, 887.95it/s]
 97%|█████████▋| 5013/5156 [00:05<00:00, 888.47it/s]
 99%|█████████▉| 5102/5156 [00:05<00:00, 886.69it/s]
100%|██████████| 5156/5156 [00:05<00:00, 931.40it/s]
2026-07-10 15:22:30.162115: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:24:43.965537: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 15:24:43.975381: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-07-10 15:24:44.706961: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:c3: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:24:44.707129: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:24:44.722306: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:24:44.722433: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:24:44.729330: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 15:24:44.735653: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 15:24:44.744027: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 15:24:44.750847: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 15:24:44.756820: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:24:44.764925: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 15:24:44.765315: 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:24:44.765407: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-07-10 15:24:44.769306: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:c3: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:24:44.769360: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:24:44.769389: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:24:44.769412: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:24:44.769434: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-07-10 15:24:44.769456: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-07-10 15:24:44.769476: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-07-10 15:24:44.769496: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-07-10 15:24:44.769517: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:24:44.774945: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-07-10 15:24:44.775005: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-07-10 15:24:46.027078: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-07-10 15:24:46.027245: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-07-10 15:24:46.027265: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-07-10 15:24:46.032286: 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:c3:00.0, compute capability: 8.0)
2026-07-10 15:24:46.083428: 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:24:46.100165: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2449915000 Hz
2026-07-10 15:24:51.381244: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-07-10 15:24:51.865248: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-07-10 15:24:51.871636: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-07-10 15:24:53.474202: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-07-10 15:24:53.619516: 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:25:27.657835: 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)