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:45.669355: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:05:07.086485: 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.166960: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-01-05 00:05:07.303628: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:05:07.303708: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:05:08.069336: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:05:08.069478: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:05:08.312278: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:05:08.882916: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:05:09.426648: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:05:09.756567: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:05:09.785022: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:05:09.790892: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:05:09.791537: 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.793929: 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.796447: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:05:09.796498: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:05:09.796529: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:05:09.796552: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:05:09.796573: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:05:09.796594: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:05:09.796613: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:05:09.796633: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:05:09.796670: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:05:09.801534: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:05:09.801604: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:05:12.411497: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-01-05 00:05:12.411615: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-01-05 00:05:12.411630: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-01-05 00:05:12.415616: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 75642 MB memory) -> physical GPU (device: 0, name: NVIDIA A100-SXM4-80GB, pci bus id: 0000:84:00.0, compute capability: 8.0)
2026-01-05 00:05:23.000513: 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.001006: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450000000 Hz
2026-01-05 00:05:24.398077: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:05:26.298839: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:05:26.304270: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:05:41.244003: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-01-05 00:05:41.328348: 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:00.596603: 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:13:34.655711: 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:39.330103: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:13:50.420001: 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:50.420985: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-01-05 00:13:50.565529: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:13:50.565604: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:13:50.572402: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:13:50.572510: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:13:50.576184: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:13:50.578601: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:13:50.583503: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:13:50.586311: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:13:50.588607: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:13:50.597326: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:13:50.597744: 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:50.600260: 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:50.602783: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:13:50.602817: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:13:50.602835: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:13:50.602850: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:13:50.602864: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:13:50.602878: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:13:50.602891: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:13:50.602904: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:13:50.602917: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:13:50.607625: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:13:50.607659: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:13:51.047906: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-01-05 00:13:51.048114: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-01-05 00:13:51.048131: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-01-05 00:13:51.053036: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 75642 MB memory) -> physical GPU (device: 0, name: NVIDIA A100-SXM4-80GB, pci bus id: 0000:84:00.0, compute capability: 8.0)
/home/users/shouvikm/miniconda3/envs/bpnet/lib/python3.7/site-packages/tensorflow/python/keras/layers/core.py:1059: UserWarning: bpnet.model.arch is not loaded, but a Lambda layer uses it. It may cause errors.
  , UserWarning)

batch:   0%|          | 0/128 [00:00<?, ?it/s]2026-01-05 00:13:52.808877: 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:52.809353: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450000000 Hz
2026-01-05 00:13:53.042415: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:13:53.481126: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:13:53.483333: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:13:54.844011: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-01-05 00:13:54.932824: 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:28<59:16, 28.00s/it]
batch:   2%|▏         | 2/128 [00:28<24:33, 11.69s/it]
batch:   2%|▏         | 3/128 [00:28<13:29,  6.48s/it]
batch:   3%|▎         | 4/128 [00:28<08:19,  4.03s/it]
batch:   4%|▍         | 5/128 [00:29<05:28,  2.67s/it]
batch:   5%|▍         | 6/128 [00:29<03:46,  1.86s/it]
batch:   5%|▌         | 7/128 [00:29<02:42,  1.34s/it]
batch:   6%|▋         | 8/128 [00:29<02:00,  1.00s/it]
batch:   7%|▋         | 9/128 [00:30<01:31,  1.29it/s]
batch:   8%|▊         | 10/128 [00:30<01:12,  1.62it/s]
batch:   9%|▊         | 11/128 [00:30<00:59,  1.95it/s]
batch:   9%|▉         | 12/128 [00:30<00:50,  2.28it/s]
batch:  10%|█         | 13/128 [00:31<00:44,  2.58it/s]
batch:  11%|█         | 14/128 [00:31<00:40,  2.84it/s]
batch:  12%|█▏        | 15/128 [00:31<00:37,  3.05it/s]
batch:  12%|█▎        | 16/128 [00:32<00:34,  3.22it/s]
batch:  13%|█▎        | 17/128 [00:32<00:33,  3.35it/s]
batch:  14%|█▍        | 18/128 [00:32<00:31,  3.46it/s]
batch:  15%|█▍        | 19/128 [00:32<00:30,  3.53it/s]
batch:  16%|█▌        | 20/128 [00:33<00:30,  3.59it/s]
batch:  16%|█▋        | 21/128 [00:33<00:29,  3.63it/s]
batch:  17%|█▋        | 22/128 [00:33<00:28,  3.66it/s]
batch:  18%|█▊        | 23/128 [00:33<00:28,  3.68it/s]
batch:  19%|█▉        | 24/128 [00:34<00:28,  3.71it/s]
batch:  20%|█▉        | 25/128 [00:34<00:27,  3.71it/s]
batch:  20%|██        | 26/128 [00:34<00:27,  3.71it/s]
batch:  21%|██        | 27/128 [00:35<00:27,  3.71it/s]
batch:  22%|██▏       | 28/128 [00:35<00:26,  3.72it/s]
batch:  23%|██▎       | 29/128 [00:35<00:26,  3.73it/s]
batch:  23%|██▎       | 30/128 [00:35<00:26,  3.72it/s]
batch:  24%|██▍       | 31/128 [00:36<00:26,  3.72it/s]
batch:  25%|██▌       | 32/128 [00:36<00:25,  3.71it/s]
batch:  26%|██▌       | 33/128 [00:36<00:25,  3.71it/s]
batch:  27%|██▋       | 34/128 [00:36<00:25,  3.72it/s]
batch:  27%|██▋       | 35/128 [00:37<00:24,  3.72it/s]
batch:  28%|██▊       | 36/128 [00:37<00:24,  3.75it/s]
batch:  29%|██▉       | 37/128 [00:37<00:24,  3.72it/s]
batch:  30%|██▉       | 38/128 [00:37<00:24,  3.70it/s]
batch:  30%|███       | 39/128 [00:38<00:24,  3.69it/s]
batch:  31%|███▏      | 40/128 [00:38<00:23,  3.69it/s]
batch:  32%|███▏      | 41/128 [00:38<00:23,  3.72it/s]
batch:  33%|███▎      | 42/128 [00:39<00:23,  3.72it/s]
batch:  34%|███▎      | 43/128 [00:39<00:22,  3.74it/s]
batch:  34%|███▍      | 44/128 [00:39<00:22,  3.73it/s]
batch:  35%|███▌      | 45/128 [00:39<00:22,  3.72it/s]
batch:  36%|███▌      | 46/128 [00:40<00:22,  3.71it/s]
batch:  37%|███▋      | 47/128 [00:40<00:21,  3.72it/s]
batch:  38%|███▊      | 48/128 [00:40<00:21,  3.72it/s]
batch:  38%|███▊      | 49/128 [00:40<00:21,  3.73it/s]
batch:  39%|███▉      | 50/128 [00:41<00:20,  3.74it/s]
batch:  40%|███▉      | 51/128 [00:41<00:20,  3.73it/s]
batch:  41%|████      | 52/128 [00:41<00:20,  3.73it/s]
batch:  41%|████▏     | 53/128 [00:42<00:20,  3.71it/s]
batch:  42%|████▏     | 54/128 [00:42<00:19,  3.70it/s]
batch:  43%|████▎     | 55/128 [00:42<00:19,  3.70it/s]
batch:  44%|████▍     | 56/128 [00:42<00:19,  3.70it/s]
batch:  45%|████▍     | 57/128 [00:43<00:19,  3.70it/s]
batch:  45%|████▌     | 58/128 [00:43<00:18,  3.70it/s]
batch:  46%|████▌     | 59/128 [00:43<00:18,  3.70it/s]
batch:  47%|████▋     | 60/128 [00:43<00:18,  3.69it/s]
batch:  48%|████▊     | 61/128 [00:44<00:18,  3.66it/s]
batch:  48%|████▊     | 62/128 [00:44<00:17,  3.69it/s]
batch:  49%|████▉     | 63/128 [00:44<00:17,  3.70it/s]
batch:  50%|█████     | 64/128 [00:45<00:17,  3.71it/s]
batch:  51%|█████     | 65/128 [00:45<00:16,  3.73it/s]
batch:  52%|█████▏    | 66/128 [00:45<00:18,  3.32it/s]
batch:  52%|█████▏    | 67/128 [00:45<00:17,  3.43it/s]
batch:  53%|█████▎    | 68/128 [00:46<00:17,  3.50it/s]
batch:  54%|█████▍    | 69/128 [00:46<00:16,  3.55it/s]
batch:  55%|█████▍    | 70/128 [00:46<00:16,  3.60it/s]
batch:  55%|█████▌    | 71/128 [00:46<00:15,  3.62it/s]
batch:  56%|█████▋    | 72/128 [00:47<00:15,  3.65it/s]
batch:  57%|█████▋    | 73/128 [00:47<00:15,  3.66it/s]
batch:  58%|█████▊    | 74/128 [00:47<00:14,  3.69it/s]
batch:  59%|█████▊    | 75/128 [00:48<00:14,  3.70it/s]
batch:  59%|█████▉    | 76/128 [00:48<00:14,  3.69it/s]
batch:  60%|██████    | 77/128 [00:48<00:13,  3.71it/s]
batch:  61%|██████    | 78/128 [00:48<00:13,  3.71it/s]
batch:  62%|██████▏   | 79/128 [00:49<00:13,  3.69it/s]
batch:  62%|██████▎   | 80/128 [00:49<00:13,  3.68it/s]
batch:  63%|██████▎   | 81/128 [00:49<00:12,  3.68it/s]
batch:  64%|██████▍   | 82/128 [00:49<00:12,  3.67it/s]
batch:  65%|██████▍   | 83/128 [00:50<00:12,  3.61it/s]
batch:  66%|██████▌   | 84/128 [00:50<00:12,  3.60it/s]
batch:  66%|██████▋   | 85/128 [00:50<00:11,  3.62it/s]
batch:  67%|██████▋   | 86/128 [00:51<00:11,  3.65it/s]
batch:  68%|██████▊   | 87/128 [00:51<00:11,  3.69it/s]
batch:  69%|██████▉   | 88/128 [00:51<00:10,  3.70it/s]
batch:  70%|██████▉   | 89/128 [00:51<00:10,  3.69it/s]
batch:  70%|███████   | 90/128 [00:52<00:10,  3.69it/s]
batch:  71%|███████   | 91/128 [00:52<00:09,  3.71it/s]
batch:  72%|███████▏  | 92/128 [00:52<00:09,  3.73it/s]
batch:  73%|███████▎  | 93/128 [00:52<00:09,  3.73it/s]
batch:  73%|███████▎  | 94/128 [00:53<00:09,  3.72it/s]
batch:  74%|███████▍  | 95/128 [00:53<00:08,  3.73it/s]
batch:  75%|███████▌  | 96/128 [00:53<00:08,  3.74it/s]
batch:  76%|███████▌  | 97/128 [00:54<00:08,  3.73it/s]
batch:  77%|███████▋  | 98/128 [00:54<00:08,  3.72it/s]
batch:  77%|███████▋  | 99/128 [00:54<00:07,  3.74it/s]
batch:  78%|███████▊  | 100/128 [00:54<00:07,  3.74it/s]
batch:  79%|███████▉  | 101/128 [00:55<00:07,  3.73it/s]
batch:  80%|███████▉  | 102/128 [00:55<00:07,  3.70it/s]
batch:  80%|████████  | 103/128 [00:55<00:06,  3.71it/s]
batch:  81%|████████▏ | 104/128 [00:55<00:06,  3.70it/s]
batch:  82%|████████▏ | 105/128 [00:56<00:06,  3.68it/s]
batch:  83%|████████▎ | 106/128 [00:56<00:05,  3.71it/s]
batch:  84%|████████▎ | 107/128 [00:56<00:05,  3.69it/s]
batch:  84%|████████▍ | 108/128 [00:57<00:05,  3.69it/s]
batch:  85%|████████▌ | 109/128 [00:57<00:05,  3.71it/s]
batch:  86%|████████▌ | 110/128 [00:57<00:04,  3.69it/s]
batch:  87%|████████▋ | 111/128 [00:57<00:04,  3.70it/s]
batch:  88%|████████▊ | 112/128 [00:58<00:04,  3.71it/s]
batch:  88%|████████▊ | 113/128 [00:58<00:04,  3.72it/s]
batch:  89%|████████▉ | 114/128 [00:58<00:03,  3.72it/s]
batch:  90%|████████▉ | 115/128 [00:58<00:03,  3.71it/s]
batch:  91%|█████████ | 116/128 [00:59<00:03,  3.70it/s]
batch:  91%|█████████▏| 117/128 [00:59<00:02,  3.70it/s]
batch:  92%|█████████▏| 118/128 [00:59<00:02,  3.70it/s]
batch:  93%|█████████▎| 119/128 [00:59<00:02,  3.69it/s]
batch:  94%|█████████▍| 120/128 [01:00<00:02,  3.67it/s]
batch:  95%|█████████▍| 121/128 [01:00<00:01,  3.70it/s]
batch:  95%|█████████▌| 122/128 [01:00<00:01,  3.67it/s]
batch:  96%|█████████▌| 123/128 [01:01<00:01,  3.70it/s]
batch:  97%|█████████▋| 124/128 [01:01<00:01,  3.70it/s]
batch:  98%|█████████▊| 125/128 [01:01<00:00,  3.70it/s]
batch:  98%|█████████▊| 126/128 [01:01<00:00,  3.70it/s]
batch:  99%|█████████▉| 127/128 [01:02<00:00,  3.71it/s]
batch: 100%|██████████| 128/128 [01:02<00:00,  4.20it/s]
batch: 100%|██████████| 128/128 [01:02<00:00,  2.05it/s]

  0%|          | 0/8160 [00:00<?, ?it/s]
  1%|▏         | 113/8160 [00:00<00:07, 1116.32it/s]
  3%|▎         | 226/8160 [00:00<00:07, 1122.94it/s]
  4%|▍         | 339/8160 [00:00<00:06, 1122.08it/s]
  6%|▌         | 452/8160 [00:00<00:06, 1124.40it/s]
  7%|▋         | 565/8160 [00:00<00:06, 1116.23it/s]
  8%|▊         | 678/8160 [00:00<00:06, 1118.26it/s]
 10%|▉         | 790/8160 [00:00<00:06, 1115.06it/s]
 11%|█         | 904/8160 [00:00<00:06, 1116.69it/s]
 12%|█▏        | 1017/8160 [00:00<00:06, 1115.60it/s]
 14%|█▍        | 1129/8160 [00:01<00:06, 1111.03it/s]
 15%|█▌        | 1241/8160 [00:01<00:06, 1112.48it/s]
 17%|█▋        | 1353/8160 [00:01<00:06, 1107.46it/s]
 18%|█▊        | 1464/8160 [00:01<00:06, 1090.31it/s]
 19%|█▉        | 1574/8160 [00:01<00:06, 1084.41it/s]
 21%|██        | 1685/8160 [00:01<00:05, 1090.46it/s]
 22%|██▏       | 1795/8160 [00:01<00:05, 1089.43it/s]
 23%|██▎       | 1906/8160 [00:01<00:05, 1095.38it/s]
 25%|██▍       | 2016/8160 [00:01<00:05, 1096.24it/s]
 26%|██▌       | 2126/8160 [00:01<00:05, 1087.81it/s]
 27%|██▋       | 2237/8160 [00:02<00:05, 1094.32it/s]
 29%|██▉       | 2347/8160 [00:02<00:05, 1088.78it/s]
 30%|███       | 2456/8160 [00:02<00:05, 1085.21it/s]
 31%|███▏      | 2566/8160 [00:02<00:05, 1085.14it/s]
 33%|███▎      | 2675/8160 [00:02<00:05, 1082.55it/s]
 34%|███▍      | 2784/8160 [00:02<00:04, 1078.77it/s]
 35%|███▌      | 2892/8160 [00:02<00:04, 1071.46it/s]
 37%|███▋      | 3000/8160 [00:02<00:04, 1073.33it/s]
 38%|███▊      | 3108/8160 [00:02<00:04, 1073.09it/s]
 39%|███▉      | 3216/8160 [00:02<00:04, 1065.45it/s]
 41%|████      | 3325/8160 [00:03<00:04, 1072.70it/s]
 42%|████▏     | 3433/8160 [00:03<00:04, 1074.69it/s]
 43%|████▎     | 3541/8160 [00:03<00:04, 1067.42it/s]
 45%|████▍     | 3650/8160 [00:03<00:04, 1068.26it/s]
 46%|████▌     | 3757/8160 [00:03<00:04, 1068.27it/s]
 47%|████▋     | 3864/8160 [00:03<00:04, 1064.90it/s]
 49%|████▊     | 3972/8160 [00:03<00:03, 1063.98it/s]
 50%|████▉     | 4079/8160 [00:03<00:03, 1038.05it/s]
 51%|█████▏    | 4186/8160 [00:03<00:03, 1045.20it/s]
 53%|█████▎    | 4292/8160 [00:03<00:03, 1046.26it/s]
 54%|█████▍    | 4398/8160 [00:04<00:03, 1049.39it/s]
 55%|█████▌    | 4505/8160 [00:04<00:03, 1055.04it/s]
 57%|█████▋    | 4611/8160 [00:04<00:03, 1050.69it/s]
 58%|█████▊    | 4719/8160 [00:04<00:03, 1054.54it/s]
 59%|█████▉    | 4825/8160 [00:04<00:03, 1053.39it/s]
 60%|██████    | 4932/8160 [00:04<00:03, 1058.07it/s]
 62%|██████▏   | 5038/8160 [00:04<00:02, 1054.37it/s]
 63%|██████▎   | 5144/8160 [00:04<00:02, 1053.54it/s]
 64%|██████▍   | 5250/8160 [00:04<00:02, 1047.41it/s]
 66%|██████▌   | 5355/8160 [00:04<00:02, 1046.25it/s]
 67%|██████▋   | 5460/8160 [00:05<00:02, 1038.28it/s]
 68%|██████▊   | 5565/8160 [00:05<00:02, 1038.82it/s]
 69%|██████▉   | 5669/8160 [00:05<00:02, 1036.24it/s]
 71%|███████   | 5773/8160 [00:05<00:02, 1034.73it/s]
 72%|███████▏  | 5878/8160 [00:05<00:02, 1039.12it/s]
 73%|███████▎  | 5982/8160 [00:05<00:02, 1036.40it/s]
 75%|███████▍  | 6086/8160 [00:05<00:02, 1036.60it/s]
 76%|███████▌  | 6190/8160 [00:05<00:01, 1036.05it/s]
 77%|███████▋  | 6294/8160 [00:05<00:01, 1034.74it/s]
 78%|███████▊  | 6398/8160 [00:05<00:01, 1028.18it/s]
 80%|███████▉  | 6502/8160 [00:06<00:01, 1027.69it/s]
 81%|████████  | 6605/8160 [00:06<00:01, 1023.62it/s]
 82%|████████▏ | 6709/8160 [00:06<00:01, 1024.11it/s]
 84%|████████▎ | 6815/8160 [00:06<00:01, 1029.51it/s]
 85%|████████▍ | 6918/8160 [00:06<00:01, 1027.94it/s]
 86%|████████▌ | 7021/8160 [00:06<00:01, 1023.96it/s]
 87%|████████▋ | 7124/8160 [00:06<00:01, 1024.83it/s]
 89%|████████▊ | 7227/8160 [00:06<00:00, 1017.02it/s]
 90%|████████▉ | 7330/8160 [00:06<00:00, 1014.82it/s]
 91%|█████████ | 7432/8160 [00:06<00:00, 1014.45it/s]
 92%|█████████▏| 7534/8160 [00:07<00:00, 1015.27it/s]
 94%|█████████▎| 7636/8160 [00:07<00:00, 1012.28it/s]
 95%|█████████▍| 7738/8160 [00:07<00:00, 1006.62it/s]
 96%|█████████▌| 7840/8160 [00:07<00:00, 1009.83it/s]
 97%|█████████▋| 7941/8160 [00:07<00:00, 1009.21it/s]
 99%|█████████▊| 8043/8160 [00:07<00:00, 1009.33it/s]
100%|█████████▉| 8144/8160 [00:07<00:00, 1006.97it/s]
100%|██████████| 8160/8160 [00:07<00:00, 1057.37it/s]

  0%|          | 0/8160 [00:00<?, ?it/s]
  1%|▏         | 110/8160 [00:00<00:07, 1082.13it/s]
  3%|▎         | 223/8160 [00:00<00:07, 1107.62it/s]
  4%|▍         | 335/8160 [00:00<00:07, 1112.28it/s]
  5%|▌         | 448/8160 [00:00<00:06, 1115.66it/s]
  7%|▋         | 560/8160 [00:00<00:06, 1113.98it/s]
  8%|▊         | 672/8160 [00:00<00:06, 1114.08it/s]
 10%|▉         | 784/8160 [00:00<00:06, 1107.12it/s]
 11%|█         | 896/8160 [00:00<00:06, 1108.26it/s]
 12%|█▏        | 1009/8160 [00:00<00:06, 1108.92it/s]
 14%|█▎        | 1121/8160 [00:01<00:06, 1107.89it/s]
 15%|█▌        | 1232/8160 [00:01<00:06, 1108.31it/s]
 16%|█▋        | 1344/8160 [00:01<00:06, 1105.74it/s]
 18%|█▊        | 1455/8160 [00:01<00:06, 1107.00it/s]
 19%|█▉        | 1566/8160 [00:01<00:05, 1104.94it/s]
 21%|██        | 1679/8160 [00:01<00:05, 1108.62it/s]
 22%|██▏       | 1790/8160 [00:01<00:05, 1108.96it/s]
 23%|██▎       | 1901/8160 [00:01<00:05, 1108.84it/s]
 25%|██▍       | 2012/8160 [00:01<00:05, 1103.83it/s]
 26%|██▌       | 2123/8160 [00:01<00:05, 1105.16it/s]
 27%|██▋       | 2234/8160 [00:02<00:05, 1102.63it/s]
 29%|██▊       | 2345/8160 [00:02<00:05, 1096.05it/s]
 30%|███       | 2455/8160 [00:02<00:05, 1092.98it/s]
 31%|███▏      | 2565/8160 [00:02<00:05, 1089.17it/s]
 33%|███▎      | 2674/8160 [00:02<00:05, 1085.47it/s]
 34%|███▍      | 2783/8160 [00:02<00:04, 1076.54it/s]
 35%|███▌      | 2892/8160 [00:02<00:04, 1074.20it/s]
 37%|███▋      | 3002/8160 [00:02<00:04, 1075.83it/s]
 38%|███▊      | 3110/8160 [00:02<00:04, 1071.70it/s]
 39%|███▉      | 3218/8160 [00:02<00:04, 1072.78it/s]
 41%|████      | 3327/8160 [00:03<00:04, 1072.67it/s]
 42%|████▏     | 3435/8160 [00:03<00:04, 1068.03it/s]
 43%|████▎     | 3542/8160 [00:03<00:04, 1068.48it/s]
 45%|████▍     | 3649/8160 [00:03<00:04, 1068.65it/s]
 46%|████▌     | 3756/8160 [00:03<00:04, 1066.43it/s]
 47%|████▋     | 3863/8160 [00:03<00:04, 1062.63it/s]
 49%|████▊     | 3970/8160 [00:03<00:03, 1060.11it/s]
 50%|████▉     | 4077/8160 [00:03<00:03, 1058.90it/s]
 51%|█████▏    | 4183/8160 [00:03<00:03, 1057.34it/s]
 53%|█████▎    | 4290/8160 [00:03<00:03, 1055.13it/s]
 54%|█████▍    | 4397/8160 [00:04<00:03, 1055.51it/s]
 55%|█████▌    | 4504/8160 [00:04<00:03, 1057.20it/s]
 57%|█████▋    | 4611/8160 [00:04<00:03, 1055.54it/s]
 58%|█████▊    | 4719/8160 [00:04<00:03, 1059.23it/s]
 59%|█████▉    | 4825/8160 [00:04<00:03, 1052.68it/s]
 60%|██████    | 4933/8160 [00:04<00:03, 1060.29it/s]
 62%|██████▏   | 5040/8160 [00:04<00:02, 1052.16it/s]
 63%|██████▎   | 5146/8160 [00:04<00:02, 1046.04it/s]
 64%|██████▍   | 5251/8160 [00:04<00:02, 1044.23it/s]
 66%|██████▌   | 5356/8160 [00:04<00:02, 1037.59it/s]
 67%|██████▋   | 5460/8160 [00:05<00:02, 1034.13it/s]
 68%|██████▊   | 5565/8160 [00:05<00:02, 1034.30it/s]
 69%|██████▉   | 5669/8160 [00:05<00:02, 1032.26it/s]
 71%|███████   | 5773/8160 [00:05<00:02, 1033.11it/s]
 72%|███████▏  | 5877/8160 [00:05<00:02, 1033.85it/s]
 73%|███████▎  | 5982/8160 [00:05<00:02, 1033.04it/s]
 75%|███████▍  | 6087/8160 [00:05<00:02, 1034.53it/s]
 76%|███████▌  | 6191/8160 [00:05<00:01, 1033.64it/s]
 77%|███████▋  | 6296/8160 [00:05<00:01, 1032.34it/s]
 78%|███████▊  | 6401/8160 [00:05<00:01, 1032.79it/s]
 80%|███████▉  | 6505/8160 [00:06<00:01, 1028.53it/s]
 81%|████████  | 6608/8160 [00:06<00:01, 1025.30it/s]
 82%|████████▏ | 6711/8160 [00:06<00:01, 1024.74it/s]
 84%|████████▎ | 6815/8160 [00:06<00:01, 1023.06it/s]
 85%|████████▍ | 6919/8160 [00:06<00:01, 1022.50it/s]
 86%|████████▌ | 7022/8160 [00:06<00:01, 1022.87it/s]
 87%|████████▋ | 7125/8160 [00:06<00:01, 1015.44it/s]
 89%|████████▊ | 7227/8160 [00:06<00:00, 1015.73it/s]
 90%|████████▉ | 7329/8160 [00:06<00:00, 1012.10it/s]
 91%|█████████ | 7431/8160 [00:06<00:00, 1007.80it/s]
 92%|█████████▏| 7532/8160 [00:07<00:00, 1007.42it/s]
 94%|█████████▎| 7634/8160 [00:07<00:00, 1010.39it/s]
 95%|█████████▍| 7736/8160 [00:07<00:00, 1011.17it/s]
 96%|█████████▌| 7838/8160 [00:07<00:00, 1010.82it/s]
 97%|█████████▋| 7940/8160 [00:07<00:00, 1010.36it/s]
 99%|█████████▊| 8042/8160 [00:07<00:00, 1009.22it/s]
100%|█████████▉| 8143/8160 [00:07<00:00, 1005.44it/s]
100%|██████████| 8160/8160 [00:07<00:00, 1057.79it/s]
2026-01-05 00:15:27.137689: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:15:33.878025: 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:33.879366: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-01-05 00:15:34.049561: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:15:34.049667: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:15:34.054156: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:15:34.054217: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:15:34.056761: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:15:34.058335: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:15:34.062069: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:15:34.064021: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:15:34.065402: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:15:34.071616: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:15:34.072088: 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:34.074371: 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:34.076916: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:84:00.0 name: NVIDIA A100-SXM4-80GB computeCapability: 8.0
coreClock: 1.41GHz coreCount: 108 deviceMemorySize: 79.14GiB deviceMemoryBandwidth: 1.85TiB/s
2026-01-05 00:15:34.076954: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:15:34.076977: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:15:34.076997: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:15:34.077015: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:15:34.077033: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:15:34.077050: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:15:34.077067: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:15:34.077084: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:15:34.081813: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:15:34.081856: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:15:34.513867: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-01-05 00:15:34.514004: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-01-05 00:15:34.514019: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-01-05 00:15:34.519506: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 75642 MB memory) -> physical GPU (device: 0, name: NVIDIA A100-SXM4-80GB, pci bus id: 0000:84:00.0, compute capability: 8.0)
2026-01-05 00:15:34.561045: 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:34.571734: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2450000000 Hz
2026-01-05 00:15:41.207730: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:15:41.598606: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:15:41.605308: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:15:42.872802: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-01-05 00:15:42.957541: 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:16:01.110025: 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)