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-01-05 00:02:46.232233: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:05:07.235616: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:05:07.237482: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-01-05 00:05:07.414335: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:07:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:05:07.414442: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:05:08.069154: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:05:08.069467: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:05:08.311569: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:05:08.882934: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:05:09.427193: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:05:09.756568: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:05:09.785028: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:05:09.791599: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:05:09.792320: 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-01-05 00:05:09.794791: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:05:09.799177: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:07:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:05:09.799230: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:05:09.799261: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:05:09.799284: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:05:09.799305: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:05:09.799326: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:05:09.799346: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:05:09.799366: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:05:09.799422: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:05:09.804341: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:05:09.804388: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:05:12.450626: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-01-05 00:05:12.450804: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-01-05 00:05:12.450821: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-01-05 00:05:12.454812: 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:07:00.0, compute capability: 8.0)
2026-01-05 00:05:23.611045: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2026-01-05 00:05:23.611659: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450000000 Hz
2026-01-05 00:05:25.004043: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:05:26.360802: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:05:26.366548: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:05:41.295369: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-01-05 00:05:41.475279: 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-01-05 00:06:13.420768: 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-01-05 00:12:56.590983: 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-01-05 00:13:01.389055: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:13:10.758481: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:13:10.759537: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-01-05 00:13:10.926057: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:07:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:13:10.926146: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:13:10.931993: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:13:10.932093: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:13:10.934966: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:13:10.936664: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:13:10.941356: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:13:10.943515: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:13:10.945226: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:13:10.951535: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:13:10.951969: 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-01-05 00:13:10.954640: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:13:10.958442: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:07:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:13:10.958492: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:13:10.958523: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:13:10.958548: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:13:10.958570: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:13:10.958592: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:13:10.958614: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:13:10.958636: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:13:10.958658: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:13:10.963474: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:13:10.963516: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:13:11.468190: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-01-05 00:13:11.468345: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-01-05 00:13:11.468361: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-01-05 00:13:11.472572: 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:07: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/128 [00:00<?, ?it/s]2026-01-05 00:13:13.480029: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2026-01-05 00:13:13.480762: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450000000 Hz
2026-01-05 00:13:13.724850: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:13:14.219949: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:13:14.222100: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:13:15.733848: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-01-05 00:13:15.866912: 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/128 [00:42<1:29:49, 42.44s/it]
batch:   2%|▏         | 2/128 [00:42<37:04, 17.65s/it]  
batch:   2%|▏         | 3/128 [00:43<20:16,  9.73s/it]
batch:   3%|▎         | 4/128 [00:43<12:25,  6.01s/it]
batch:   4%|▍         | 5/128 [00:43<08:05,  3.95s/it]
batch:   5%|▍         | 6/128 [00:43<05:30,  2.71s/it]
batch:   5%|▌         | 7/128 [00:44<03:52,  1.92s/it]
batch:   6%|▋         | 8/128 [00:44<02:48,  1.40s/it]
batch:   7%|▋         | 9/128 [00:44<02:05,  1.05s/it]
batch:   8%|▊         | 10/128 [00:45<01:36,  1.22it/s]
batch:   9%|▊         | 11/128 [00:45<01:17,  1.51it/s]
batch:   9%|▉         | 12/128 [00:45<01:03,  1.81it/s]
batch:  10%|█         | 13/128 [00:46<00:54,  2.09it/s]
batch:  11%|█         | 14/128 [00:46<00:48,  2.36it/s]
batch:  12%|█▏        | 15/128 [00:46<00:43,  2.59it/s]
batch:  12%|█▎        | 16/128 [00:46<00:40,  2.77it/s]
batch:  13%|█▎        | 17/128 [00:47<00:38,  2.91it/s]
batch:  14%|█▍        | 18/128 [00:47<00:36,  3.01it/s]
batch:  15%|█▍        | 19/128 [00:47<00:35,  3.10it/s]
batch:  16%|█▌        | 20/128 [00:48<00:34,  3.16it/s]
batch:  16%|█▋        | 21/128 [00:48<00:33,  3.22it/s]
batch:  17%|█▋        | 22/128 [00:48<00:32,  3.27it/s]
batch:  18%|█▊        | 23/128 [00:49<00:31,  3.32it/s]
batch:  19%|█▉        | 24/128 [00:49<00:31,  3.33it/s]
batch:  20%|█▉        | 25/128 [00:49<00:30,  3.34it/s]
batch:  20%|██        | 26/128 [00:49<00:30,  3.39it/s]
batch:  21%|██        | 27/128 [00:50<00:29,  3.38it/s]
batch:  22%|██▏       | 28/128 [00:50<00:29,  3.37it/s]
batch:  23%|██▎       | 29/128 [00:50<00:29,  3.39it/s]
batch:  23%|██▎       | 30/128 [00:51<00:28,  3.40it/s]
batch:  24%|██▍       | 31/128 [00:51<00:28,  3.42it/s]
batch:  25%|██▌       | 32/128 [00:51<00:28,  3.38it/s]
batch:  26%|██▌       | 33/128 [00:51<00:28,  3.37it/s]
batch:  27%|██▋       | 34/128 [00:52<00:27,  3.37it/s]
batch:  27%|██▋       | 35/128 [00:52<00:27,  3.40it/s]
batch:  28%|██▊       | 36/128 [00:52<00:26,  3.43it/s]
batch:  29%|██▉       | 37/128 [00:53<00:26,  3.43it/s]
batch:  30%|██▉       | 38/128 [00:53<00:26,  3.45it/s]
batch:  30%|███       | 39/128 [00:53<00:25,  3.49it/s]
batch:  31%|███▏      | 40/128 [00:53<00:24,  3.53it/s]
batch:  32%|███▏      | 41/128 [00:54<00:24,  3.53it/s]
batch:  33%|███▎      | 42/128 [00:54<00:24,  3.50it/s]
batch:  34%|███▎      | 43/128 [00:54<00:24,  3.44it/s]
batch:  34%|███▍      | 44/128 [00:55<00:24,  3.45it/s]
batch:  35%|███▌      | 45/128 [00:55<00:23,  3.47it/s]
batch:  36%|███▌      | 46/128 [00:55<00:23,  3.46it/s]
batch:  37%|███▋      | 47/128 [00:55<00:23,  3.52it/s]
batch:  38%|███▊      | 48/128 [00:56<00:22,  3.54it/s]
batch:  38%|███▊      | 49/128 [00:56<00:22,  3.53it/s]
batch:  39%|███▉      | 50/128 [00:56<00:22,  3.54it/s]
batch:  40%|███▉      | 51/128 [00:57<00:21,  3.53it/s]
batch:  41%|████      | 52/128 [00:57<00:21,  3.53it/s]
batch:  41%|████▏     | 53/128 [00:57<00:21,  3.53it/s]
batch:  42%|████▏     | 54/128 [00:57<00:20,  3.53it/s]
batch:  43%|████▎     | 55/128 [00:58<00:20,  3.52it/s]
batch:  44%|████▍     | 56/128 [00:58<00:20,  3.52it/s]
batch:  45%|████▍     | 57/128 [00:58<00:20,  3.51it/s]
batch:  45%|████▌     | 58/128 [00:59<00:19,  3.54it/s]
batch:  46%|████▌     | 59/128 [00:59<00:19,  3.55it/s]
batch:  47%|████▋     | 60/128 [00:59<00:19,  3.55it/s]
batch:  48%|████▊     | 61/128 [00:59<00:18,  3.55it/s]
batch:  48%|████▊     | 62/128 [01:00<00:18,  3.54it/s]
batch:  49%|████▉     | 63/128 [01:00<00:18,  3.53it/s]
batch:  50%|█████     | 64/128 [01:00<00:18,  3.48it/s]
batch:  51%|█████     | 65/128 [01:01<00:18,  3.43it/s]
batch:  52%|█████▏    | 66/128 [01:01<00:20,  3.04it/s]
batch:  52%|█████▏    | 67/128 [01:01<00:19,  3.14it/s]
batch:  53%|█████▎    | 68/128 [01:02<00:18,  3.24it/s]
batch:  54%|█████▍    | 69/128 [01:02<00:17,  3.31it/s]
batch:  55%|█████▍    | 70/128 [01:02<00:17,  3.39it/s]
batch:  55%|█████▌    | 71/128 [01:02<00:16,  3.42it/s]
batch:  56%|█████▋    | 72/128 [01:03<00:16,  3.46it/s]
batch:  57%|█████▋    | 73/128 [01:03<00:15,  3.46it/s]
batch:  58%|█████▊    | 74/128 [01:03<00:15,  3.43it/s]
batch:  59%|█████▊    | 75/128 [01:04<00:15,  3.46it/s]
batch:  59%|█████▉    | 76/128 [01:04<00:15,  3.44it/s]
batch:  60%|██████    | 77/128 [01:04<00:14,  3.43it/s]
batch:  61%|██████    | 78/128 [01:05<00:14,  3.39it/s]
batch:  62%|██████▏   | 79/128 [01:05<00:14,  3.37it/s]
batch:  62%|██████▎   | 80/128 [01:05<00:14,  3.42it/s]
batch:  63%|██████▎   | 81/128 [01:05<00:13,  3.40it/s]
batch:  64%|██████▍   | 82/128 [01:06<00:13,  3.39it/s]
batch:  65%|██████▍   | 83/128 [01:06<00:13,  3.42it/s]
batch:  66%|██████▌   | 84/128 [01:06<00:12,  3.44it/s]
batch:  66%|██████▋   | 85/128 [01:07<00:12,  3.46it/s]
batch:  67%|██████▋   | 86/128 [01:07<00:12,  3.48it/s]
batch:  68%|██████▊   | 87/128 [01:07<00:11,  3.51it/s]
batch:  69%|██████▉   | 88/128 [01:07<00:11,  3.51it/s]
batch:  70%|██████▉   | 89/128 [01:08<00:11,  3.51it/s]
batch:  70%|███████   | 90/128 [01:08<00:10,  3.51it/s]
batch:  71%|███████   | 91/128 [01:08<00:10,  3.51it/s]
batch:  72%|███████▏  | 92/128 [01:09<00:10,  3.51it/s]
batch:  73%|███████▎  | 93/128 [01:09<00:09,  3.53it/s]
batch:  73%|███████▎  | 94/128 [01:09<00:09,  3.53it/s]
batch:  74%|███████▍  | 95/128 [01:09<00:09,  3.53it/s]
batch:  75%|███████▌  | 96/128 [01:10<00:09,  3.51it/s]
batch:  76%|███████▌  | 97/128 [01:10<00:08,  3.48it/s]
batch:  77%|███████▋  | 98/128 [01:10<00:08,  3.49it/s]
batch:  77%|███████▋  | 99/128 [01:11<00:08,  3.52it/s]
batch:  78%|███████▊  | 100/128 [01:11<00:07,  3.54it/s]
batch:  79%|███████▉  | 101/128 [01:11<00:07,  3.56it/s]
batch:  80%|███████▉  | 102/128 [01:11<00:07,  3.57it/s]
batch:  80%|████████  | 103/128 [01:12<00:06,  3.58it/s]
batch:  81%|████████▏ | 104/128 [01:12<00:06,  3.58it/s]
batch:  82%|████████▏ | 105/128 [01:12<00:06,  3.56it/s]
batch:  83%|████████▎ | 106/128 [01:12<00:06,  3.57it/s]
batch:  84%|████████▎ | 107/128 [01:13<00:05,  3.57it/s]
batch:  84%|████████▍ | 108/128 [01:13<00:05,  3.57it/s]
batch:  85%|████████▌ | 109/128 [01:13<00:05,  3.59it/s]
batch:  86%|████████▌ | 110/128 [01:14<00:05,  3.57it/s]
batch:  87%|████████▋ | 111/128 [01:14<00:04,  3.56it/s]
batch:  88%|████████▊ | 112/128 [01:14<00:04,  3.55it/s]
batch:  88%|████████▊ | 113/128 [01:14<00:04,  3.55it/s]
batch:  89%|████████▉ | 114/128 [01:15<00:03,  3.56it/s]
batch:  90%|████████▉ | 115/128 [01:15<00:03,  3.53it/s]
batch:  91%|█████████ | 116/128 [01:15<00:03,  3.54it/s]
batch:  91%|█████████▏| 117/128 [01:16<00:03,  3.53it/s]
batch:  92%|█████████▏| 118/128 [01:16<00:02,  3.50it/s]
batch:  93%|█████████▎| 119/128 [01:16<00:02,  3.51it/s]
batch:  94%|█████████▍| 120/128 [01:16<00:02,  3.53it/s]
batch:  95%|█████████▍| 121/128 [01:17<00:01,  3.52it/s]
batch:  95%|█████████▌| 122/128 [01:17<00:01,  3.52it/s]
batch:  96%|█████████▌| 123/128 [01:17<00:01,  3.50it/s]
batch:  97%|█████████▋| 124/128 [01:18<00:01,  3.51it/s]
batch:  98%|█████████▊| 125/128 [01:18<00:00,  3.51it/s]
batch:  98%|█████████▊| 126/128 [01:18<00:00,  3.54it/s]
batch:  99%|█████████▉| 127/128 [01:18<00:00,  3.52it/s]
batch: 100%|██████████| 128/128 [01:19<00:00,  4.00it/s]
batch: 100%|██████████| 128/128 [01:19<00:00,  1.62it/s]

  0%|          | 0/8160 [00:00<?, ?it/s]
  1%|▏         | 110/8160 [00:00<00:07, 1097.84it/s]
  3%|▎         | 220/8160 [00:00<00:07, 1089.24it/s]
  4%|▍         | 329/8160 [00:00<00:07, 1088.10it/s]
  5%|▌         | 438/8160 [00:00<00:07, 1079.18it/s]
  7%|▋         | 546/8160 [00:00<00:07, 1079.20it/s]
  8%|▊         | 654/8160 [00:00<00:06, 1078.45it/s]
  9%|▉         | 762/8160 [00:00<00:06, 1069.62it/s]
 11%|█         | 872/8160 [00:00<00:06, 1074.89it/s]
 12%|█▏        | 981/8160 [00:00<00:06, 1073.01it/s]
 13%|█▎        | 1094/8160 [00:01<00:06, 1084.34it/s]
 15%|█▍        | 1203/8160 [00:01<00:06, 1074.53it/s]
 16%|█▌        | 1313/8160 [00:01<00:06, 1081.66it/s]
 17%|█▋        | 1422/8160 [00:01<00:06, 1080.81it/s]
 19%|█▉        | 1531/8160 [00:01<00:06, 1067.25it/s]
 20%|██        | 1641/8160 [00:01<00:06, 1075.67it/s]
 21%|██▏       | 1750/8160 [00:01<00:05, 1079.21it/s]
 23%|██▎       | 1858/8160 [00:01<00:05, 1072.56it/s]
 24%|██▍       | 1966/8160 [00:01<00:05, 1061.34it/s]
 25%|██▌       | 2073/8160 [00:01<00:05, 1039.61it/s]
 27%|██▋       | 2178/8160 [00:02<00:05, 1042.59it/s]
 28%|██▊       | 2287/8160 [00:02<00:05, 1052.97it/s]
 29%|██▉       | 2393/8160 [00:02<00:05, 1044.13it/s]
 31%|███       | 2498/8160 [00:02<00:05, 1037.72it/s]
 32%|███▏      | 2604/8160 [00:02<00:05, 1043.07it/s]
 33%|███▎      | 2709/8160 [00:02<00:05, 1044.77it/s]
 34%|███▍      | 2814/8160 [00:02<00:05, 1038.95it/s]
 36%|███▌      | 2919/8160 [00:02<00:05, 1040.50it/s]
 37%|███▋      | 3024/8160 [00:02<00:04, 1035.54it/s]
 38%|███▊      | 3128/8160 [00:02<00:04, 1035.16it/s]
 40%|███▉      | 3232/8160 [00:03<00:04, 1036.45it/s]
 41%|████      | 3336/8160 [00:03<00:04, 1030.46it/s]
 42%|████▏     | 3440/8160 [00:03<00:04, 1028.42it/s]
 43%|████▎     | 3547/8160 [00:03<00:04, 1038.96it/s]
 45%|████▍     | 3651/8160 [00:03<00:04, 1032.25it/s]
 46%|████▌     | 3755/8160 [00:03<00:04, 1030.74it/s]
 47%|████▋     | 3859/8160 [00:03<00:04, 1027.74it/s]
 49%|████▊     | 3962/8160 [00:03<00:04, 1021.47it/s]
 50%|████▉     | 4065/8160 [00:03<00:04, 1019.23it/s]
 51%|█████     | 4169/8160 [00:03<00:03, 1024.57it/s]
 52%|█████▏    | 4272/8160 [00:04<00:03, 1022.51it/s]
 54%|█████▎    | 4375/8160 [00:04<00:03, 1024.20it/s]
 55%|█████▍    | 4482/8160 [00:04<00:03, 1032.86it/s]
 56%|█████▌    | 4586/8160 [00:04<00:03, 1003.58it/s]
 57%|█████▊    | 4692/8160 [00:04<00:03, 1014.33it/s]
 59%|█████▉    | 4794/8160 [00:04<00:03, 1007.72it/s]
 60%|█████▉    | 4895/8160 [00:04<00:03, 1006.50it/s]
 61%|██████▏   | 5000/8160 [00:04<00:03, 1018.26it/s]
 63%|██████▎   | 5102/8160 [00:04<00:03, 1010.63it/s]
 64%|██████▍   | 5204/8160 [00:04<00:02, 999.32it/s] 
 65%|██████▌   | 5304/8160 [00:05<00:02, 990.68it/s]
 66%|██████▌   | 5404/8160 [00:05<00:02, 988.68it/s]
 67%|██████▋   | 5504/8160 [00:05<00:02, 990.02it/s]
 69%|██████▊   | 5606/8160 [00:05<00:02, 998.22it/s]
 70%|██████▉   | 5706/8160 [00:05<00:02, 995.14it/s]
 71%|███████   | 5807/8160 [00:05<00:02, 999.12it/s]
 72%|███████▏  | 5909/8160 [00:05<00:02, 1004.38it/s]
 74%|███████▎  | 6010/8160 [00:05<00:02, 996.47it/s] 
 75%|███████▍  | 6114/8160 [00:05<00:02, 1004.59it/s]
 76%|███████▌  | 6215/8160 [00:06<00:01, 1004.37it/s]
 77%|███████▋  | 6316/8160 [00:06<00:01, 998.97it/s] 
 79%|███████▊  | 6420/8160 [00:06<00:01, 1006.13it/s]
 80%|███████▉  | 6521/8160 [00:06<00:01, 994.10it/s] 
 81%|████████  | 6622/8160 [00:06<00:01, 997.80it/s]
 82%|████████▏ | 6725/8160 [00:06<00:01, 1006.45it/s]
 84%|████████▎ | 6826/8160 [00:06<00:01, 993.18it/s] 
 85%|████████▍ | 6927/8160 [00:06<00:01, 992.22it/s]
 86%|████████▌ | 7027/8160 [00:06<00:01, 992.83it/s]
 87%|████████▋ | 7127/8160 [00:06<00:01, 977.96it/s]
 89%|████████▊ | 7228/8160 [00:07<00:00, 984.10it/s]
 90%|████████▉ | 7327/8160 [00:07<00:00, 983.25it/s]
 91%|█████████ | 7426/8160 [00:07<00:00, 973.43it/s]
 92%|█████████▏| 7527/8160 [00:07<00:00, 981.58it/s]
 93%|█████████▎| 7626/8160 [00:07<00:00, 962.07it/s]
 95%|█████████▍| 7724/8160 [00:07<00:00, 963.37it/s]
 96%|█████████▌| 7823/8160 [00:07<00:00, 969.19it/s]
 97%|█████████▋| 7920/8160 [00:07<00:00, 947.96it/s]
 98%|█████████▊| 8015/8160 [00:07<00:00, 939.29it/s]
 99%|█████████▉| 8110/8160 [00:07<00:00, 941.19it/s]
100%|██████████| 8160/8160 [00:08<00:00, 1018.62it/s]

  0%|          | 0/8160 [00:00<?, ?it/s]
  1%|▏         | 106/8160 [00:00<00:07, 1059.87it/s]
  3%|▎         | 212/8160 [00:00<00:07, 1041.33it/s]
  4%|▍         | 322/8160 [00:00<00:07, 1066.23it/s]
  5%|▌         | 429/8160 [00:00<00:07, 1051.33it/s]
  7%|▋         | 535/8160 [00:00<00:07, 1035.80it/s]
  8%|▊         | 639/8160 [00:00<00:07, 1023.80it/s]
  9%|▉         | 742/8160 [00:00<00:07, 1020.51it/s]
 10%|█         | 846/8160 [00:00<00:07, 1026.56it/s]
 12%|█▏        | 954/8160 [00:00<00:06, 1036.57it/s]
 13%|█▎        | 1058/8160 [00:01<00:06, 1035.10it/s]
 14%|█▍        | 1170/8160 [00:01<00:06, 1055.56it/s]
 16%|█▌        | 1281/8160 [00:01<00:06, 1069.26it/s]
 17%|█▋        | 1392/8160 [00:01<00:06, 1080.84it/s]
 18%|█▊        | 1501/8160 [00:01<00:06, 1079.31it/s]
 20%|█▉        | 1609/8160 [00:01<00:06, 1053.91it/s]
 21%|██        | 1715/8160 [00:01<00:06, 1049.27it/s]
 22%|██▏       | 1821/8160 [00:01<00:06, 1031.20it/s]
 24%|██▎       | 1925/8160 [00:01<00:06, 1022.76it/s]
 25%|██▍       | 2028/8160 [00:01<00:06, 1019.72it/s]
 26%|██▌       | 2131/8160 [00:02<00:05, 1020.41it/s]
 27%|██▋       | 2239/8160 [00:02<00:05, 1037.22it/s]
 29%|██▊       | 2343/8160 [00:02<00:05, 1034.59it/s]
 30%|███       | 2448/8160 [00:02<00:05, 1034.01it/s]
 31%|███▏      | 2552/8160 [00:02<00:05, 1034.27it/s]
 33%|███▎      | 2656/8160 [00:02<00:05, 1031.61it/s]
 34%|███▍      | 2760/8160 [00:02<00:05, 1033.18it/s]
 35%|███▌      | 2864/8160 [00:02<00:05, 1029.94it/s]
 36%|███▋      | 2968/8160 [00:02<00:05, 1027.56it/s]
 38%|███▊      | 3074/8160 [00:02<00:04, 1036.23it/s]
 39%|███▉      | 3178/8160 [00:03<00:04, 1023.25it/s]
 40%|████      | 3282/8160 [00:03<00:04, 1025.18it/s]
 42%|████▏     | 3389/8160 [00:03<00:04, 1034.99it/s]
 43%|████▎     | 3493/8160 [00:03<00:04, 1026.87it/s]
 44%|████▍     | 3596/8160 [00:03<00:04, 1017.81it/s]
 45%|████▌     | 3700/8160 [00:03<00:04, 1022.04it/s]
 47%|████▋     | 3803/8160 [00:03<00:04, 1014.52it/s]
 48%|████▊     | 3905/8160 [00:03<00:04, 1013.59it/s]
 49%|████▉     | 4008/8160 [00:03<00:04, 1012.28it/s]
 50%|█████     | 4110/8160 [00:03<00:04, 1000.08it/s]
 52%|█████▏    | 4213/8160 [00:04<00:03, 1003.62it/s]
 53%|█████▎    | 4314/8160 [00:04<00:03, 1001.79it/s]
 54%|█████▍    | 4415/8160 [00:04<00:03, 1002.63it/s]
 55%|█████▌    | 4521/8160 [00:04<00:03, 1013.69it/s]
 57%|█████▋    | 4625/8160 [00:04<00:03, 1016.55it/s]
 58%|█████▊    | 4727/8160 [00:04<00:03, 1010.87it/s]
 59%|█████▉    | 4831/8160 [00:04<00:03, 1017.84it/s]
 60%|██████    | 4933/8160 [00:04<00:03, 1016.75it/s]
 62%|██████▏   | 5035/8160 [00:04<00:03, 1010.08it/s]
 63%|██████▎   | 5137/8160 [00:04<00:03, 1006.68it/s]
 64%|██████▍   | 5238/8160 [00:05<00:02, 994.36it/s] 
 65%|██████▌   | 5339/8160 [00:05<00:02, 995.23it/s]
 67%|██████▋   | 5443/8160 [00:05<00:02, 1004.20it/s]
 68%|██████▊   | 5544/8160 [00:05<00:02, 997.57it/s] 
 69%|██████▉   | 5646/8160 [00:05<00:02, 1004.13it/s]
 70%|███████   | 5747/8160 [00:05<00:02, 1001.06it/s]
 72%|███████▏  | 5848/8160 [00:05<00:02, 994.38it/s] 
 73%|███████▎  | 5948/8160 [00:05<00:02, 994.08it/s]
 74%|███████▍  | 6048/8160 [00:05<00:02, 994.64it/s]
 75%|███████▌  | 6150/8160 [00:06<00:02, 997.85it/s]
 77%|███████▋  | 6255/8160 [00:06<00:01, 1008.46it/s]
 78%|███████▊  | 6356/8160 [00:06<00:01, 1005.30it/s]
 79%|███████▉  | 6457/8160 [00:06<00:01, 1005.20it/s]
 80%|████████  | 6559/8160 [00:06<00:01, 1005.20it/s]
 82%|████████▏ | 6661/8160 [00:06<00:01, 1009.33it/s]
 83%|████████▎ | 6762/8160 [00:06<00:01, 1004.18it/s]
 84%|████████▍ | 6863/8160 [00:06<00:01, 1001.56it/s]
 85%|████████▌ | 6964/8160 [00:06<00:01, 999.19it/s] 
 87%|████████▋ | 7064/8160 [00:06<00:01, 987.14it/s]
 88%|████████▊ | 7166/8160 [00:07<00:01, 993.14it/s]
 89%|████████▉ | 7267/8160 [00:07<00:00, 993.70it/s]
 90%|█████████ | 7367/8160 [00:07<00:00, 988.26it/s]
 92%|█████████▏| 7467/8160 [00:07<00:00, 985.57it/s]
 93%|█████████▎| 7566/8160 [00:07<00:00, 972.08it/s]
 94%|█████████▍| 7664/8160 [00:07<00:00, 968.30it/s]
 95%|█████████▌| 7766/8160 [00:07<00:00, 981.06it/s]
 96%|█████████▋| 7865/8160 [00:07<00:00, 975.92it/s]
 98%|█████████▊| 7965/8160 [00:07<00:00, 978.68it/s]
 99%|█████████▉| 8065/8160 [00:07<00:00, 981.93it/s]
100%|██████████| 8160/8160 [00:08<00:00, 1013.81it/s]
2026-01-05 00:15:10.253543: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:15:19.464735: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:15:19.472351: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-01-05 00:15:19.646445: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:07:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:15:19.646561: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:15:19.653098: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:15:19.653206: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:15:19.655969: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:15:19.657777: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:15:19.662685: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:15:19.664928: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:15:19.666675: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:15:19.672985: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:15:19.673640: 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-01-05 00:15:19.676239: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:15:19.680156: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:07:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:15:19.680251: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:15:19.680293: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:15:19.680330: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:15:19.680363: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:15:19.680395: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:15:19.680427: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:15:19.680459: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:15:19.680490: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:15:19.685331: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:15:19.685381: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:15:20.146567: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-01-05 00:15:20.146765: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-01-05 00:15:20.146798: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-01-05 00:15:20.153031: 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:07:00.0, compute capability: 8.0)
2026-01-05 00:15:20.199485: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:196] None of the MLIR optimization passes are enabled (registered 0 passes)
2026-01-05 00:15:20.213210: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450000000 Hz
2026-01-05 00:15:27.237614: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:15:27.679761: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:15:27.686673: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:15:29.267790: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-01-05 00:15:29.406261: 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-01-05 00:15:56.027803: 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)