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-12-29 21:52:45.051581: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:53:04.465306: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-12-29 21:53:04.473093: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2025-12-29 21:53:04.696206: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:8a:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2025-12-29 21:53:04.696348: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:53:05.273854: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-12-29 21:53:05.274108: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-12-29 21:53:05.649204: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-12-29 21:53:06.206548: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-12-29 21:53:06.856242: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-12-29 21:53:07.119616: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-12-29 21:53:07.243518: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-12-29 21:53:07.248842: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-12-29 21:53:07.249504: 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.
2025-12-29 21:53:07.252072: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-12-29 21:53:07.254782: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:8a:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2025-12-29 21:53:07.254835: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:53:07.254863: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-12-29 21:53:07.254884: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-12-29 21:53:07.254903: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-12-29 21:53:07.254921: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-12-29 21:53:07.254940: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-12-29 21:53:07.254958: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-12-29 21:53:07.255005: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-12-29 21:53:07.261136: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-12-29 21:53:07.261183: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:53:09.853774: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2025-12-29 21:53:09.853947: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2025-12-29 21:53:09.853964: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2025-12-29 21:53:09.859081: 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:8a:00.0, compute capability: 8.0)
2025-12-29 21:53:24.730583: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2025-12-29 21:53:24.731100: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450010000 Hz
2025-12-29 21:53:26.414414: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-12-29 21:53:28.379441: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-12-29 21:53:28.385965: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-12-29 21:53:42.332455: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2025-12-29 21:53:42.510942: 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-12-29 21:54:23.705253: 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-12-29 21:57:39.383957: 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-12-29 21:57:46.284724: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:58:05.222847: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-12-29 21:58:05.224389: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2025-12-29 21:58:05.471407: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:8a:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2025-12-29 21:58:05.471548: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:58:05.480050: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-12-29 21:58:05.480152: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-12-29 21:58:05.484749: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-12-29 21:58:05.487784: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-12-29 21:58:05.494001: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-12-29 21:58:05.496435: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-12-29 21:58:05.498934: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-12-29 21:58:05.504269: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-12-29 21:58:05.504829: 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.
2025-12-29 21:58:05.508041: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-12-29 21:58:05.512165: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:8a:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2025-12-29 21:58:05.512258: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:58:05.512310: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-12-29 21:58:05.512328: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-12-29 21:58:05.512355: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-12-29 21:58:05.512373: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-12-29 21:58:05.512389: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-12-29 21:58:05.512405: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-12-29 21:58:05.512422: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-12-29 21:58:05.516474: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-12-29 21:58:05.516526: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:58:06.116652: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2025-12-29 21:58:06.116829: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2025-12-29 21:58:06.116845: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2025-12-29 21:58:06.120865: 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:8a: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/54 [00:00<?, ?it/s]2025-12-29 21:58:08.430612: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2025-12-29 21:58:08.432005: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450010000 Hz
2025-12-29 21:58:08.715876: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-12-29 21:58:09.275059: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-12-29 21:58:09.277549: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-12-29 21:58:10.955680: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2025-12-29 21:58:11.173217: 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:   2%|▏         | 1/54 [00:43<38:39, 43.76s/it]
batch:   4%|▎         | 2/54 [00:44<15:47, 18.21s/it]
batch:   6%|▌         | 3/54 [00:44<08:32, 10.05s/it]
batch:   7%|▋         | 4/54 [00:44<05:10,  6.20s/it]
batch:   9%|▉         | 5/54 [00:45<03:20,  4.08s/it]
batch:  11%|█         | 6/54 [00:45<02:14,  2.80s/it]
batch:  13%|█▎        | 7/54 [00:45<01:33,  1.99s/it]
batch:  15%|█▍        | 8/54 [00:46<01:07,  1.46s/it]
batch:  17%|█▋        | 9/54 [00:46<00:49,  1.10s/it]
batch:  19%|█▊        | 10/54 [00:46<00:37,  1.17it/s]
batch:  20%|██        | 11/54 [00:46<00:29,  1.44it/s]
batch:  22%|██▏       | 12/54 [00:47<00:24,  1.72it/s]
batch:  24%|██▍       | 13/54 [00:47<00:20,  1.99it/s]
batch:  26%|██▌       | 14/54 [00:47<00:17,  2.24it/s]
batch:  28%|██▊       | 15/54 [00:48<00:15,  2.45it/s]
batch:  30%|██▉       | 16/54 [00:48<00:14,  2.64it/s]
batch:  31%|███▏      | 17/54 [00:48<00:13,  2.75it/s]
batch:  33%|███▎      | 18/54 [00:49<00:13,  2.76it/s]
batch:  35%|███▌      | 19/54 [00:49<00:12,  2.85it/s]
batch:  37%|███▋      | 20/54 [00:49<00:11,  2.91it/s]
batch:  39%|███▉      | 21/54 [00:50<00:11,  2.95it/s]
batch:  41%|████      | 22/54 [00:50<00:10,  3.00it/s]
batch:  43%|████▎     | 23/54 [00:50<00:10,  3.01it/s]
batch:  44%|████▍     | 24/54 [00:51<00:09,  3.03it/s]
batch:  46%|████▋     | 25/54 [00:51<00:09,  3.03it/s]
batch:  48%|████▊     | 26/54 [00:51<00:09,  3.05it/s]
batch:  50%|█████     | 27/54 [00:52<00:08,  3.05it/s]
batch:  52%|█████▏    | 28/54 [00:52<00:08,  3.09it/s]
batch:  54%|█████▎    | 29/54 [00:52<00:08,  3.10it/s]
batch:  56%|█████▌    | 30/54 [00:53<00:07,  3.13it/s]
batch:  57%|█████▋    | 31/54 [00:53<00:07,  3.11it/s]
batch:  59%|█████▉    | 32/54 [00:53<00:07,  3.13it/s]
batch:  61%|██████    | 33/54 [00:54<00:06,  3.15it/s]
batch:  63%|██████▎   | 34/54 [00:54<00:06,  3.08it/s]
batch:  65%|██████▍   | 35/54 [00:54<00:06,  3.09it/s]
batch:  67%|██████▋   | 36/54 [00:55<00:05,  3.10it/s]
batch:  69%|██████▊   | 37/54 [00:55<00:05,  3.02it/s]
batch:  70%|███████   | 38/54 [00:55<00:05,  3.05it/s]
batch:  72%|███████▏  | 39/54 [00:56<00:04,  3.08it/s]
batch:  74%|███████▍  | 40/54 [00:56<00:04,  3.09it/s]
batch:  76%|███████▌  | 41/54 [00:56<00:04,  3.11it/s]
batch:  78%|███████▊  | 42/54 [00:57<00:03,  3.09it/s]
batch:  80%|███████▉  | 43/54 [00:57<00:03,  3.09it/s]
batch:  81%|████████▏ | 44/54 [00:57<00:03,  3.08it/s]
batch:  83%|████████▎ | 45/54 [00:57<00:02,  3.08it/s]
batch:  85%|████████▌ | 46/54 [00:58<00:02,  3.10it/s]
batch:  87%|████████▋ | 47/54 [00:58<00:02,  3.13it/s]
batch:  89%|████████▉ | 48/54 [00:58<00:01,  3.12it/s]
batch:  91%|█████████ | 49/54 [00:59<00:01,  3.10it/s]
batch:  93%|█████████▎| 50/54 [00:59<00:01,  3.08it/s]
batch:  94%|█████████▍| 51/54 [00:59<00:00,  3.10it/s]
batch:  96%|█████████▋| 52/54 [01:00<00:00,  3.08it/s]
batch:  98%|█████████▊| 53/54 [01:00<00:00,  3.64it/s]
batch: 100%|██████████| 54/54 [01:00<00:00,  1.12s/it]

  0%|          | 0/3350 [00:00<?, ?it/s]
  3%|▎         | 99/3350 [00:00<00:03, 982.28it/s]
  6%|▌         | 198/3350 [00:00<00:03, 975.86it/s]
  9%|▉         | 296/3350 [00:00<00:03, 941.16it/s]
 12%|█▏        | 397/3350 [00:00<00:03, 960.15it/s]
 15%|█▍        | 496/3350 [00:00<00:02, 966.61it/s]
 18%|█▊        | 595/3350 [00:00<00:02, 973.66it/s]
 21%|██        | 693/3350 [00:00<00:02, 963.87it/s]
 24%|██▎       | 790/3350 [00:00<00:02, 964.09it/s]
 26%|██▋       | 887/3350 [00:00<00:02, 962.44it/s]
 29%|██▉       | 984/3350 [00:01<00:02, 954.37it/s]
 32%|███▏      | 1080/3350 [00:01<00:02, 942.71it/s]
 35%|███▌      | 1175/3350 [00:01<00:02, 943.14it/s]
 38%|███▊      | 1270/3350 [00:01<00:02, 944.71it/s]
 41%|████      | 1365/3350 [00:01<00:02, 933.09it/s]
 44%|████▎     | 1459/3350 [00:01<00:02, 931.90it/s]
 46%|████▋     | 1553/3350 [00:01<00:01, 925.85it/s]
 49%|████▉     | 1646/3350 [00:01<00:01, 925.53it/s]
 52%|█████▏    | 1739/3350 [00:01<00:01, 922.26it/s]
 55%|█████▍    | 1832/3350 [00:01<00:01, 914.21it/s]
 57%|█████▋    | 1924/3350 [00:02<00:01, 915.75it/s]
 60%|██████    | 2016/3350 [00:02<00:01, 913.95it/s]
 63%|██████▎   | 2108/3350 [00:02<00:01, 902.41it/s]
 66%|██████▌   | 2199/3350 [00:02<00:01, 904.05it/s]
 68%|██████▊   | 2290/3350 [00:02<00:01, 901.90it/s]
 71%|███████   | 2381/3350 [00:02<00:01, 900.03it/s]
 74%|███████▍  | 2472/3350 [00:02<00:00, 891.15it/s]
 76%|███████▋  | 2562/3350 [00:02<00:00, 889.74it/s]
 79%|███████▉  | 2651/3350 [00:02<00:00, 879.63it/s]
 82%|████████▏ | 2741/3350 [00:02<00:00, 882.67it/s]
 84%|████████▍ | 2830/3350 [00:03<00:00, 883.89it/s]
 87%|████████▋ | 2919/3350 [00:03<00:00, 877.97it/s]
 90%|████████▉ | 3007/3350 [00:03<00:00, 876.50it/s]
 92%|█████████▏| 3095/3350 [00:03<00:00, 870.39it/s]
 95%|█████████▌| 3183/3350 [00:03<00:00, 871.59it/s]
 98%|█████████▊| 3271/3350 [00:03<00:00, 868.90it/s]
100%|██████████| 3350/3350 [00:03<00:00, 913.81it/s]

  0%|          | 0/3350 [00:00<?, ?it/s]
  3%|▎         | 96/3350 [00:00<00:03, 955.96it/s]
  6%|▌         | 195/3350 [00:00<00:03, 965.76it/s]
  9%|▊         | 292/3350 [00:00<00:03, 965.34it/s]
 12%|█▏        | 391/3350 [00:00<00:03, 969.58it/s]
 15%|█▍        | 491/3350 [00:00<00:02, 976.21it/s]
 18%|█▊        | 589/3350 [00:00<00:02, 947.92it/s]
 20%|██        | 685/3350 [00:00<00:02, 949.46it/s]
 23%|██▎       | 782/3350 [00:00<00:02, 953.37it/s]
 26%|██▌       | 878/3350 [00:00<00:02, 954.67it/s]
 29%|██▉       | 974/3350 [00:01<00:02, 947.37it/s]
 32%|███▏      | 1069/3350 [00:01<00:02, 938.15it/s]
 35%|███▍      | 1164/3350 [00:01<00:02, 939.99it/s]
 38%|███▊      | 1259/3350 [00:01<00:02, 942.74it/s]
 40%|████      | 1354/3350 [00:01<00:02, 924.12it/s]
 43%|████▎     | 1448/3350 [00:01<00:02, 924.57it/s]
 46%|████▌     | 1543/3350 [00:01<00:01, 925.87it/s]
 49%|████▉     | 1637/3350 [00:01<00:01, 926.23it/s]
 52%|█████▏    | 1730/3350 [00:01<00:01, 922.99it/s]
 54%|█████▍    | 1823/3350 [00:01<00:01, 923.93it/s]
 57%|█████▋    | 1916/3350 [00:02<00:01, 924.45it/s]
 60%|█████▉    | 2009/3350 [00:02<00:01, 895.77it/s]
 63%|██████▎   | 2100/3350 [00:02<00:01, 896.54it/s]
 65%|██████▌   | 2191/3350 [00:02<00:01, 900.27it/s]
 68%|██████▊   | 2282/3350 [00:02<00:01, 903.09it/s]
 71%|███████   | 2373/3350 [00:02<00:01, 875.25it/s]
 74%|███████▎  | 2463/3350 [00:02<00:01, 881.10it/s]
 76%|███████▌  | 2552/3350 [00:02<00:00, 872.02it/s]
 79%|███████▉  | 2641/3350 [00:02<00:00, 872.85it/s]
 82%|████████▏ | 2731/3350 [00:02<00:00, 874.86it/s]
 84%|████████▍ | 2819/3350 [00:03<00:00, 873.71it/s]
 87%|████████▋ | 2907/3350 [00:03<00:00, 830.65it/s]
 89%|████████▉ | 2995/3350 [00:03<00:00, 840.57it/s]
 92%|█████████▏| 3082/3350 [00:03<00:00, 847.30it/s]
 95%|█████████▍| 3169/3350 [00:03<00:00, 852.10it/s]
 97%|█████████▋| 3256/3350 [00:03<00:00, 856.16it/s]
100%|█████████▉| 3343/3350 [00:03<00:00, 859.38it/s]
100%|██████████| 3350/3350 [00:03<00:00, 904.39it/s]
2025-12-29 21:59:33.176475: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:59:40.610995: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-12-29 21:59:40.612691: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2025-12-29 21:59:40.852576: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:8a:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2025-12-29 21:59:40.852764: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:59:40.857476: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-12-29 21:59:40.857561: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-12-29 21:59:40.859752: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-12-29 21:59:40.861008: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-12-29 21:59:40.865110: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-12-29 21:59:40.866935: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-12-29 21:59:40.868121: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-12-29 21:59:40.875295: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-12-29 21:59:40.875859: 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.
2025-12-29 21:59:40.878438: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2025-12-29 21:59:40.881086: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:8a:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2025-12-29 21:59:40.881170: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:59:40.881222: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-12-29 21:59:40.881267: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-12-29 21:59:40.881310: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2025-12-29 21:59:40.881386: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2025-12-29 21:59:40.881434: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2025-12-29 21:59:40.881492: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2025-12-29 21:59:40.881534: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-12-29 21:59:40.887200: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2025-12-29 21:59:40.887251: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2025-12-29 21:59:41.434453: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2025-12-29 21:59:41.434623: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2025-12-29 21:59:41.434639: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2025-12-29 21:59:41.439527: 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:8a:00.0, compute capability: 8.0)
2025-12-29 21:59:41.487231: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:196] None of the MLIR optimization passes are enabled (registered 0 passes)
2025-12-29 21:59:41.500111: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450010000 Hz
2025-12-29 21:59:44.990418: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2025-12-29 21:59:45.521172: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2025-12-29 21:59:45.526597: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2025-12-29 21:59:47.291940: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2025-12-29 21:59:47.431227: 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-12-29 22:00:15.125973: 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)