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:40:01.359240: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:41:42.763581: 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:42.793322: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-05-28 11:41:43.092494: 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-05-28 11:41:43.092611: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:41:43.714946: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:41:43.715130: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:41:44.160241: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:41:44.663486: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:41:45.217616: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:41:45.453932: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:41:45.704389: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:41:45.714248: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:41:45.714768: 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-05-28 11:41:45.715049: 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:45.717685: 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-05-28 11:41:45.717731: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:41:45.717758: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:41:45.717772: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:41:45.717785: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:41:45.717799: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:41:45.717812: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:41:45.717825: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:41:45.717857: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:41:45.722669: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:41:45.722714: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:41:48.424330: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-05-28 11:41:48.424434: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-05-28 11:41:48.424451: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-05-28 11:41:48.429206: 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-05-28 11:41:56.953380: 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:56.953886: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2449805000 Hz
2026-05-28 11:42:00.531527: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:42:03.185358: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:42:03.190200: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:42:19.606126: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-05-28 11:42:19.730092: 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:45.216655: 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:53:56.583756: 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:54:00.735167: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:54:07.793547: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-05-28 11:54:07.794758: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-05-28 11:54:08.123863: 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-05-28 11:54:08.123953: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:54:08.129312: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:54:08.129389: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:54:08.131765: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:54:08.133341: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:54:08.137562: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:54:08.139466: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:54:08.141072: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:54:08.147373: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:54:08.147891: 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-05-28 11:54:08.148146: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-05-28 11:54:08.150649: 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-05-28 11:54:08.150689: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:54:08.150714: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:54:08.150734: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:54:08.150751: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:54:08.150767: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:54:08.150784: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:54:08.150803: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:54:08.150821: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:54:08.156733: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:54:08.156779: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:54:08.656423: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-05-28 11:54:08.656498: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-05-28 11:54:08.656514: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-05-28 11:54:08.660792: 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/80 [00:00<?, ?it/s]2026-05-28 11:54:10.680307: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2026-05-28 11:54:10.680800: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2449805000 Hz
2026-05-28 11:54:10.925024: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:54:11.440105: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:54:11.441885: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:54:12.972766: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-05-28 11:54:13.098760: 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:28<37:51, 28.75s/it]
batch:   2%|▎         | 2/80 [00:29<15:37, 12.01s/it]
batch:   4%|▍         | 3/80 [00:29<08:33,  6.66s/it]
batch:   5%|▌         | 4/80 [00:29<05:15,  4.15s/it]
batch:   6%|▋         | 5/80 [00:29<03:27,  2.76s/it]
batch:   8%|▊         | 6/80 [00:30<02:22,  1.92s/it]
batch:   9%|▉         | 7/80 [00:30<01:41,  1.39s/it]
batch:  10%|█         | 8/80 [00:30<01:15,  1.04s/it]
batch:  11%|█▏        | 9/80 [00:31<00:57,  1.24it/s]
batch:  12%|█▎        | 10/80 [00:31<00:45,  1.54it/s]
batch:  14%|█▍        | 11/80 [00:31<00:37,  1.85it/s]
batch:  15%|█▌        | 12/80 [00:32<00:31,  2.14it/s]
batch:  16%|█▋        | 13/80 [00:32<00:27,  2.42it/s]
batch:  18%|█▊        | 14/80 [00:32<00:24,  2.65it/s]
batch:  19%|█▉        | 15/80 [00:32<00:22,  2.83it/s]
batch:  20%|██        | 16/80 [00:33<00:21,  2.98it/s]
batch:  21%|██▏       | 17/80 [00:33<00:20,  3.09it/s]
batch:  22%|██▎       | 18/80 [00:33<00:19,  3.17it/s]
batch:  24%|██▍       | 19/80 [00:34<00:18,  3.22it/s]
batch:  25%|██▌       | 20/80 [00:34<00:18,  3.26it/s]
batch:  26%|██▋       | 21/80 [00:34<00:17,  3.30it/s]
batch:  28%|██▊       | 22/80 [00:34<00:17,  3.33it/s]
batch:  29%|██▉       | 23/80 [00:35<00:17,  3.34it/s]
batch:  30%|███       | 24/80 [00:35<00:16,  3.36it/s]
batch:  31%|███▏      | 25/80 [00:35<00:16,  3.37it/s]
batch:  32%|███▎      | 26/80 [00:36<00:15,  3.38it/s]
batch:  34%|███▍      | 27/80 [00:36<00:15,  3.37it/s]
batch:  35%|███▌      | 28/80 [00:36<00:15,  3.38it/s]
batch:  36%|███▋      | 29/80 [00:37<00:15,  3.37it/s]
batch:  38%|███▊      | 30/80 [00:37<00:14,  3.38it/s]
batch:  39%|███▉      | 31/80 [00:37<00:14,  3.38it/s]
batch:  40%|████      | 32/80 [00:37<00:14,  3.38it/s]
batch:  41%|████▏     | 33/80 [00:38<00:13,  3.39it/s]
batch:  42%|████▎     | 34/80 [00:38<00:13,  3.38it/s]
batch:  44%|████▍     | 35/80 [00:38<00:13,  3.38it/s]
batch:  45%|████▌     | 36/80 [00:39<00:13,  3.38it/s]
batch:  46%|████▋     | 37/80 [00:39<00:12,  3.37it/s]
batch:  48%|████▊     | 38/80 [00:39<00:12,  3.39it/s]
batch:  49%|████▉     | 39/80 [00:39<00:12,  3.36it/s]
batch:  50%|█████     | 40/80 [00:40<00:11,  3.37it/s]
batch:  51%|█████▏    | 41/80 [00:40<00:11,  3.37it/s]
batch:  52%|█████▎    | 42/80 [00:40<00:11,  3.38it/s]
batch:  54%|█████▍    | 43/80 [00:41<00:10,  3.39it/s]
batch:  55%|█████▌    | 44/80 [00:41<00:10,  3.38it/s]
batch:  56%|█████▋    | 45/80 [00:41<00:10,  3.39it/s]
batch:  57%|█████▊    | 46/80 [00:42<00:10,  3.39it/s]
batch:  59%|█████▉    | 47/80 [00:42<00:09,  3.40it/s]
batch:  60%|██████    | 48/80 [00:42<00:09,  3.40it/s]
batch:  61%|██████▏   | 49/80 [00:42<00:09,  3.39it/s]
batch:  62%|██████▎   | 50/80 [00:43<00:08,  3.40it/s]
batch:  64%|██████▍   | 51/80 [00:43<00:08,  3.38it/s]
batch:  65%|██████▌   | 52/80 [00:43<00:08,  3.39it/s]
batch:  66%|██████▋   | 53/80 [00:44<00:07,  3.39it/s]
batch:  68%|██████▊   | 54/80 [00:44<00:07,  3.38it/s]
batch:  69%|██████▉   | 55/80 [00:44<00:07,  3.38it/s]
batch:  70%|███████   | 56/80 [00:45<00:07,  3.39it/s]
batch:  71%|███████▏  | 57/80 [00:45<00:06,  3.40it/s]
batch:  72%|███████▎  | 58/80 [00:45<00:06,  3.40it/s]
batch:  74%|███████▍  | 59/80 [00:45<00:06,  3.39it/s]
batch:  75%|███████▌  | 60/80 [00:46<00:05,  3.39it/s]
batch:  76%|███████▋  | 61/80 [00:46<00:05,  3.38it/s]
batch:  78%|███████▊  | 62/80 [00:46<00:05,  3.38it/s]
batch:  79%|███████▉  | 63/80 [00:47<00:05,  3.38it/s]
batch:  80%|████████  | 64/80 [00:47<00:04,  3.39it/s]
batch:  81%|████████▏ | 65/80 [00:47<00:04,  3.39it/s]
batch:  82%|████████▎ | 66/80 [00:48<00:04,  2.99it/s]
batch:  84%|████████▍ | 67/80 [00:48<00:04,  3.08it/s]
batch:  85%|████████▌ | 68/80 [00:48<00:03,  3.16it/s]
batch:  86%|████████▋ | 69/80 [00:48<00:03,  3.22it/s]
batch:  88%|████████▊ | 70/80 [00:49<00:03,  3.28it/s]
batch:  89%|████████▉ | 71/80 [00:49<00:02,  3.29it/s]
batch:  90%|█████████ | 72/80 [00:49<00:02,  3.33it/s]
batch:  91%|█████████▏| 73/80 [00:50<00:02,  3.34it/s]
batch:  92%|█████████▎| 74/80 [00:50<00:01,  3.32it/s]
batch:  94%|█████████▍| 75/80 [00:50<00:01,  3.34it/s]
batch:  95%|█████████▌| 76/80 [00:51<00:01,  3.34it/s]
batch:  96%|█████████▋| 77/80 [00:51<00:00,  3.36it/s]
batch:  98%|█████████▊| 78/80 [00:51<00:00,  3.36it/s]
batch:  99%|█████████▉| 79/80 [00:51<00:00,  3.37it/s]
batch: 100%|██████████| 80/80 [00:52<00:00,  3.80it/s]
batch: 100%|██████████| 80/80 [00:52<00:00,  1.53it/s]

  0%|          | 0/5091 [00:00<?, ?it/s]
  2%|▏         | 99/5091 [00:00<00:05, 987.64it/s]
  4%|▍         | 198/5091 [00:00<00:05, 977.80it/s]
  6%|▌         | 297/5091 [00:00<00:04, 982.74it/s]
  8%|▊         | 396/5091 [00:00<00:04, 984.19it/s]
 10%|▉         | 495/5091 [00:00<00:04, 980.14it/s]
 12%|█▏        | 594/5091 [00:00<00:04, 981.34it/s]
 14%|█▎        | 693/5091 [00:00<00:04, 980.53it/s]
 16%|█▌        | 792/5091 [00:00<00:04, 980.63it/s]
 18%|█▊        | 891/5091 [00:00<00:04, 979.23it/s]
 19%|█▉        | 990/5091 [00:01<00:04, 978.56it/s]
 21%|██▏       | 1088/5091 [00:01<00:04, 978.42it/s]
 23%|██▎       | 1187/5091 [00:01<00:04, 975.89it/s]
 25%|██▌       | 1285/5091 [00:01<00:03, 971.76it/s]
 27%|██▋       | 1383/5091 [00:01<00:03, 973.76it/s]
 29%|██▉       | 1481/5091 [00:01<00:03, 970.37it/s]
 31%|███       | 1579/5091 [00:01<00:03, 963.35it/s]
 33%|███▎      | 1676/5091 [00:01<00:03, 957.97it/s]
 35%|███▍      | 1773/5091 [00:01<00:03, 956.30it/s]
 37%|███▋      | 1869/5091 [00:01<00:03, 956.18it/s]
 39%|███▊      | 1965/5091 [00:02<00:03, 952.24it/s]
 40%|████      | 2061/5091 [00:02<00:03, 949.35it/s]
 42%|████▏     | 2157/5091 [00:02<00:03, 948.07it/s]
 44%|████▍     | 2252/5091 [00:02<00:02, 947.92it/s]
 46%|████▌     | 2348/5091 [00:02<00:02, 945.23it/s]
 48%|████▊     | 2443/5091 [00:02<00:02, 944.42it/s]
 50%|████▉     | 2538/5091 [00:02<00:02, 940.00it/s]
 52%|█████▏    | 2634/5091 [00:02<00:02, 940.47it/s]
 54%|█████▎    | 2729/5091 [00:02<00:02, 940.49it/s]
 55%|█████▌    | 2824/5091 [00:02<00:02, 936.34it/s]
 57%|█████▋    | 2918/5091 [00:03<00:02, 931.84it/s]
 59%|█████▉    | 3012/5091 [00:03<00:02, 931.36it/s]
 61%|██████    | 3106/5091 [00:03<00:02, 932.18it/s]
 63%|██████▎   | 3200/5091 [00:03<00:02, 927.29it/s]
 65%|██████▍   | 3294/5091 [00:03<00:01, 925.24it/s]
 67%|██████▋   | 3388/5091 [00:03<00:01, 923.68it/s]
 68%|██████▊   | 3481/5091 [00:03<00:01, 919.73it/s]
 70%|███████   | 3574/5091 [00:03<00:01, 917.20it/s]
 72%|███████▏  | 3666/5091 [00:03<00:01, 913.20it/s]
 74%|███████▍  | 3759/5091 [00:03<00:01, 917.74it/s]
 76%|███████▌  | 3851/5091 [00:04<00:01, 913.09it/s]
 77%|███████▋  | 3943/5091 [00:04<00:01, 910.17it/s]
 79%|███████▉  | 4035/5091 [00:04<00:01, 912.19it/s]
 81%|████████  | 4127/5091 [00:04<00:01, 901.83it/s]
 83%|████████▎ | 4218/5091 [00:04<00:00, 899.13it/s]
 85%|████████▍ | 4308/5091 [00:04<00:00, 897.85it/s]
 86%|████████▋ | 4399/5091 [00:04<00:00, 898.94it/s]
 88%|████████▊ | 4490/5091 [00:04<00:00, 899.60it/s]
 90%|████████▉ | 4580/5091 [00:04<00:00, 897.46it/s]
 92%|█████████▏| 4670/5091 [00:04<00:00, 893.07it/s]
 94%|█████████▎| 4761/5091 [00:05<00:00, 893.20it/s]
 95%|█████████▌| 4851/5091 [00:05<00:00, 889.75it/s]
 97%|█████████▋| 4941/5091 [00:05<00:00, 889.11it/s]
 99%|█████████▉| 5031/5091 [00:05<00:00, 889.05it/s]
100%|██████████| 5091/5091 [00:05<00:00, 934.45it/s]

  0%|          | 0/5091 [00:00<?, ?it/s]
  2%|▏         | 98/5091 [00:00<00:05, 977.31it/s]
  4%|▍         | 196/5091 [00:00<00:05, 977.08it/s]
  6%|▌         | 295/5091 [00:00<00:04, 974.57it/s]
  8%|▊         | 394/5091 [00:00<00:04, 979.52it/s]
 10%|▉         | 493/5091 [00:00<00:04, 983.07it/s]
 12%|█▏        | 592/5091 [00:00<00:04, 984.04it/s]
 14%|█▎        | 691/5091 [00:00<00:04, 981.91it/s]
 16%|█▌        | 790/5091 [00:00<00:04, 980.50it/s]
 17%|█▋        | 889/5091 [00:00<00:04, 979.05it/s]
 19%|█▉        | 987/5091 [00:01<00:04, 975.84it/s]
 21%|██▏       | 1085/5091 [00:01<00:04, 973.32it/s]
 23%|██▎       | 1184/5091 [00:01<00:04, 972.77it/s]
 25%|██▌       | 1283/5091 [00:01<00:03, 971.77it/s]
 27%|██▋       | 1381/5091 [00:01<00:03, 968.06it/s]
 29%|██▉       | 1479/5091 [00:01<00:03, 966.75it/s]
 31%|███       | 1576/5091 [00:01<00:03, 964.36it/s]
 33%|███▎      | 1673/5091 [00:01<00:03, 964.53it/s]
 35%|███▍      | 1770/5091 [00:01<00:03, 960.58it/s]
 37%|███▋      | 1867/5091 [00:01<00:03, 953.63it/s]
 39%|███▊      | 1963/5091 [00:02<00:03, 947.91it/s]
 40%|████      | 2060/5091 [00:02<00:03, 948.78it/s]
 42%|████▏     | 2157/5091 [00:02<00:03, 949.35it/s]
 44%|████▍     | 2252/5091 [00:02<00:02, 948.30it/s]
 46%|████▌     | 2347/5091 [00:02<00:02, 948.66it/s]
 48%|████▊     | 2442/5091 [00:02<00:02, 941.68it/s]
 50%|████▉     | 2537/5091 [00:02<00:02, 938.15it/s]
 52%|█████▏    | 2633/5091 [00:02<00:02, 944.28it/s]
 54%|█████▎    | 2728/5091 [00:02<00:02, 937.28it/s]
 55%|█████▌    | 2822/5091 [00:02<00:02, 936.89it/s]
 57%|█████▋    | 2916/5091 [00:03<00:02, 932.19it/s]
 59%|█████▉    | 3010/5091 [00:03<00:02, 931.81it/s]
 61%|██████    | 3104/5091 [00:03<00:02, 924.69it/s]
 63%|██████▎   | 3197/5091 [00:03<00:02, 924.86it/s]
 65%|██████▍   | 3291/5091 [00:03<00:01, 924.38it/s]
 66%|██████▋   | 3384/5091 [00:03<00:01, 925.67it/s]
 68%|██████▊   | 3477/5091 [00:03<00:01, 921.02it/s]
 70%|███████   | 3570/5091 [00:03<00:01, 917.32it/s]
 72%|███████▏  | 3662/5091 [00:03<00:01, 913.74it/s]
 74%|███████▎  | 3754/5091 [00:03<00:01, 909.91it/s]
 76%|███████▌  | 3846/5091 [00:04<00:01, 912.46it/s]
 77%|███████▋  | 3938/5091 [00:04<00:01, 909.59it/s]
 79%|███████▉  | 4029/5091 [00:04<00:01, 909.13it/s]
 81%|████████  | 4120/5091 [00:04<00:01, 907.19it/s]
 83%|████████▎ | 4211/5091 [00:04<00:00, 902.57it/s]
 85%|████████▍ | 4302/5091 [00:04<00:00, 901.82it/s]
 86%|████████▋ | 4393/5091 [00:04<00:00, 896.75it/s]
 88%|████████▊ | 4484/5091 [00:04<00:00, 897.30it/s]
 90%|████████▉ | 4574/5091 [00:04<00:00, 895.46it/s]
 92%|█████████▏| 4665/5091 [00:04<00:00, 894.34it/s]
 93%|█████████▎| 4755/5091 [00:05<00:00, 892.12it/s]
 95%|█████████▌| 4845/5091 [00:05<00:00, 893.23it/s]
 97%|█████████▋| 4935/5091 [00:05<00:00, 890.40it/s]
 99%|█████████▊| 5025/5091 [00:05<00:00, 881.25it/s]
100%|██████████| 5091/5091 [00:05<00:00, 933.45it/s]
2026-05-28 11:55:31.058121: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:55:39.559707: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-05-28 11:55:39.560876: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-05-28 11:55:39.926320: 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-05-28 11:55:39.926422: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:55:39.933527: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:55:39.933610: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:55:39.936483: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:55:39.939673: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:55:39.943645: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:55:39.945841: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:55:39.947097: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:55:39.952799: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:55:39.953254: 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-05-28 11:55:39.953518: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-05-28 11:55:39.955878: 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-05-28 11:55:39.955975: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:55:39.956010: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:55:39.956037: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:55:39.956061: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-05-28 11:55:39.956086: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-05-28 11:55:39.956109: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-05-28 11:55:39.956132: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-05-28 11:55:39.956156: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:55:39.962228: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-05-28 11:55:39.962279: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-05-28 11:55:40.448588: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-05-28 11:55:40.448682: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-05-28 11:55:40.448697: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-05-28 11:55:40.453450: 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-05-28 11:55:40.498947: 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:55:40.511678: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2449805000 Hz
2026-05-28 11:55:45.666992: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-05-28 11:55:46.121124: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-05-28 11:55:46.126242: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-05-28 11:55:47.524059: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-05-28 11:55:47.645183: 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:56:11.049404: 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)