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:05:34.027866: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:06:21.306852: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:06:21.321831: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-01-05 00:06:21.399787: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-01-05 00:06:21.399837: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:06:21.965601: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:06:21.965668: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:06:22.375721: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:06:22.840036: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:06:23.466210: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:06:23.662330: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:06:23.840411: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:06:23.840808: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:06:23.841161: 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 AVX512F FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2026-01-05 00:06:23.843029: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:06:23.843319: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-01-05 00:06:23.843345: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:06:23.843359: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:06:23.843369: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:06:23.843379: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:06:23.843388: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:06:23.843397: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:06:23.843406: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:06:23.843426: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:06:23.843717: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:06:23.843738: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:06:26.338968: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-01-05 00:06:26.339062: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-01-05 00:06:26.339073: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-01-05 00:06:26.339760: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 42313 MB memory) -> physical GPU (device: 0, name: NVIDIA L40S, pci bus id: 0000:61:00.0, compute capability: 8.9)
2026-01-05 00:06:38.616214: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2026-01-05 00:06:38.616821: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2500000000 Hz
2026-01-05 00:06:39.757341: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:06:41.548610: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:06:41.555927: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:06:55.160413: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-01-05 00:06:55.253342: 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:07:25.123702: 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:19:45.932636: 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:20:08.394898: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:20:26.399817: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:20:26.400590: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-01-05 00:20:26.429832: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-01-05 00:20:26.429878: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:20:26.439440: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:20:26.439523: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:20:26.444054: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:20:26.448535: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:20:26.454477: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:20:26.458069: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:20:26.460898: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:20:26.461273: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:20:26.461483: 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 AVX512F FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2026-01-05 00:20:26.463036: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:20:26.463280: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-01-05 00:20:26.463302: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:20:26.463316: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:20:26.463326: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:20:26.463336: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:20:26.463347: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:20:26.463356: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:20:26.463366: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:20:26.463376: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:20:26.463662: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:20:26.463684: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:20:26.878863: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-01-05 00:20:26.878958: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-01-05 00:20:26.878968: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-01-05 00:20:26.879634: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 42313 MB memory) -> physical GPU (device: 0, name: NVIDIA L40S, pci bus id: 0000:61:00.0, compute capability: 8.9)
/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:20:28.420275: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2)
2026-01-05 00:20:28.420858: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2500000000 Hz
2026-01-05 00:20:28.627944: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:20:29.146858: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:20:29.149631: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:20:30.767910: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-01-05 00:20:30.872630: 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:31<1:06:02, 31.20s/it]
batch:   2%|▏         | 2/128 [00:31<27:17, 12.99s/it]  
batch:   2%|▏         | 3/128 [00:31<14:55,  7.17s/it]
batch:   3%|▎         | 4/128 [00:31<09:09,  4.43s/it]
batch:   4%|▍         | 5/128 [00:32<05:58,  2.92s/it]
batch:   5%|▍         | 6/128 [00:32<04:04,  2.01s/it]
batch:   5%|▌         | 7/128 [00:32<02:52,  1.43s/it]
batch:   6%|▋         | 8/128 [00:32<02:05,  1.05s/it]
batch:   7%|▋         | 9/128 [00:33<01:34,  1.26it/s]
batch:   8%|▊         | 10/128 [00:33<01:13,  1.61it/s]
batch:   9%|▊         | 11/128 [00:33<00:58,  1.99it/s]
batch:   9%|▉         | 12/128 [00:33<00:48,  2.37it/s]
batch:  10%|█         | 13/128 [00:34<00:41,  2.75it/s]
batch:  11%|█         | 14/128 [00:34<00:37,  3.08it/s]
batch:  12%|█▏        | 15/128 [00:34<00:33,  3.33it/s]
batch:  12%|█▎        | 16/128 [00:34<00:31,  3.57it/s]
batch:  13%|█▎        | 17/128 [00:34<00:29,  3.73it/s]
batch:  14%|█▍        | 18/128 [00:35<00:28,  3.85it/s]
batch:  15%|█▍        | 19/128 [00:35<00:27,  3.95it/s]
batch:  16%|█▌        | 20/128 [00:35<00:27,  3.99it/s]
batch:  16%|█▋        | 21/128 [00:35<00:26,  4.04it/s]
batch:  17%|█▋        | 22/128 [00:36<00:26,  4.05it/s]
batch:  18%|█▊        | 23/128 [00:36<00:25,  4.04it/s]
batch:  19%|█▉        | 24/128 [00:36<00:25,  4.07it/s]
batch:  20%|█▉        | 25/128 [00:36<00:25,  4.08it/s]
batch:  20%|██        | 26/128 [00:37<00:25,  4.08it/s]
batch:  21%|██        | 27/128 [00:37<00:24,  4.06it/s]
batch:  22%|██▏       | 28/128 [00:37<00:24,  4.08it/s]
batch:  23%|██▎       | 29/128 [00:37<00:24,  4.12it/s]
batch:  23%|██▎       | 30/128 [00:38<00:23,  4.14it/s]
batch:  24%|██▍       | 31/128 [00:38<00:23,  4.17it/s]
batch:  25%|██▌       | 32/128 [00:38<00:23,  4.16it/s]
batch:  26%|██▌       | 33/128 [00:38<00:22,  4.18it/s]
batch:  27%|██▋       | 34/128 [00:39<00:22,  4.18it/s]
batch:  27%|██▋       | 35/128 [00:39<00:22,  4.18it/s]
batch:  28%|██▊       | 36/128 [00:39<00:21,  4.19it/s]
batch:  29%|██▉       | 37/128 [00:39<00:21,  4.16it/s]
batch:  30%|██▉       | 38/128 [00:40<00:21,  4.18it/s]
batch:  30%|███       | 39/128 [00:40<00:21,  4.16it/s]
batch:  31%|███▏      | 40/128 [00:40<00:21,  4.16it/s]
batch:  32%|███▏      | 41/128 [00:40<00:20,  4.18it/s]
batch:  33%|███▎      | 42/128 [00:41<00:20,  4.20it/s]
batch:  34%|███▎      | 43/128 [00:41<00:20,  4.21it/s]
batch:  34%|███▍      | 44/128 [00:41<00:19,  4.23it/s]
batch:  35%|███▌      | 45/128 [00:41<00:19,  4.24it/s]
batch:  36%|███▌      | 46/128 [00:41<00:19,  4.23it/s]
batch:  37%|███▋      | 47/128 [00:42<00:19,  4.25it/s]
batch:  38%|███▊      | 48/128 [00:42<00:18,  4.25it/s]
batch:  38%|███▊      | 49/128 [00:42<00:18,  4.25it/s]
batch:  39%|███▉      | 50/128 [00:42<00:18,  4.25it/s]
batch:  40%|███▉      | 51/128 [00:43<00:18,  4.26it/s]
batch:  41%|████      | 52/128 [00:43<00:17,  4.25it/s]
batch:  41%|████▏     | 53/128 [00:43<00:17,  4.20it/s]
batch:  42%|████▏     | 54/128 [00:43<00:17,  4.19it/s]
batch:  43%|████▎     | 55/128 [00:44<00:17,  4.22it/s]
batch:  44%|████▍     | 56/128 [00:44<00:17,  4.23it/s]
batch:  45%|████▍     | 57/128 [00:44<00:16,  4.25it/s]
batch:  45%|████▌     | 58/128 [00:44<00:16,  4.25it/s]
batch:  46%|████▌     | 59/128 [00:45<00:16,  4.24it/s]
batch:  47%|████▋     | 60/128 [00:45<00:16,  4.24it/s]
batch:  48%|████▊     | 61/128 [00:45<00:15,  4.24it/s]
batch:  48%|████▊     | 62/128 [00:45<00:15,  4.24it/s]
batch:  49%|████▉     | 63/128 [00:45<00:15,  4.24it/s]
batch:  50%|█████     | 64/128 [00:46<00:15,  4.23it/s]
batch:  51%|█████     | 65/128 [00:46<00:14,  4.22it/s]
batch:  52%|█████▏    | 66/128 [00:46<00:17,  3.60it/s]
batch:  52%|█████▏    | 67/128 [00:47<00:16,  3.76it/s]
batch:  53%|█████▎    | 68/128 [00:47<00:15,  3.89it/s]
batch:  54%|█████▍    | 69/128 [00:47<00:14,  3.98it/s]
batch:  55%|█████▍    | 70/128 [00:47<00:14,  4.04it/s]
batch:  55%|█████▌    | 71/128 [00:48<00:14,  4.07it/s]
batch:  56%|█████▋    | 72/128 [00:48<00:13,  4.12it/s]
batch:  57%|█████▋    | 73/128 [00:48<00:13,  4.13it/s]
batch:  58%|█████▊    | 74/128 [00:48<00:12,  4.15it/s]
batch:  59%|█████▊    | 75/128 [00:48<00:12,  4.17it/s]
batch:  59%|█████▉    | 76/128 [00:49<00:12,  4.14it/s]
batch:  60%|██████    | 77/128 [00:49<00:12,  4.17it/s]
batch:  61%|██████    | 78/128 [00:49<00:11,  4.18it/s]
batch:  62%|██████▏   | 79/128 [00:49<00:11,  4.17it/s]
batch:  62%|██████▎   | 80/128 [00:50<00:11,  4.17it/s]
batch:  63%|██████▎   | 81/128 [00:50<00:11,  4.18it/s]
batch:  64%|██████▍   | 82/128 [00:50<00:10,  4.21it/s]
batch:  65%|██████▍   | 83/128 [00:50<00:10,  4.22it/s]
batch:  66%|██████▌   | 84/128 [00:51<00:10,  4.22it/s]
batch:  66%|██████▋   | 85/128 [00:51<00:10,  4.22it/s]
batch:  67%|██████▋   | 86/128 [00:51<00:09,  4.22it/s]
batch:  68%|██████▊   | 87/128 [00:51<00:09,  4.23it/s]
batch:  69%|██████▉   | 88/128 [00:52<00:09,  4.22it/s]
batch:  70%|██████▉   | 89/128 [00:52<00:09,  4.24it/s]
batch:  70%|███████   | 90/128 [00:52<00:08,  4.25it/s]
batch:  71%|███████   | 91/128 [00:52<00:08,  4.25it/s]
batch:  72%|███████▏  | 92/128 [00:52<00:08,  4.23it/s]
batch:  73%|███████▎  | 93/128 [00:53<00:08,  4.20it/s]
batch:  73%|███████▎  | 94/128 [00:53<00:08,  4.22it/s]
batch:  74%|███████▍  | 95/128 [00:53<00:07,  4.23it/s]
batch:  75%|███████▌  | 96/128 [00:53<00:07,  4.23it/s]
batch:  76%|███████▌  | 97/128 [00:54<00:07,  4.25it/s]
batch:  77%|███████▋  | 98/128 [00:54<00:07,  4.22it/s]
batch:  77%|███████▋  | 99/128 [00:54<00:06,  4.24it/s]
batch:  78%|███████▊  | 100/128 [00:54<00:06,  4.21it/s]
batch:  79%|███████▉  | 101/128 [00:55<00:06,  4.23it/s]
batch:  80%|███████▉  | 102/128 [00:55<00:06,  4.25it/s]
batch:  80%|████████  | 103/128 [00:55<00:05,  4.23it/s]
batch:  81%|████████▏ | 104/128 [00:55<00:05,  4.22it/s]
batch:  82%|████████▏ | 105/128 [00:56<00:05,  4.19it/s]
batch:  83%|████████▎ | 106/128 [00:56<00:05,  4.18it/s]
batch:  84%|████████▎ | 107/128 [00:56<00:05,  4.18it/s]
batch:  84%|████████▍ | 108/128 [00:56<00:04,  4.19it/s]
batch:  85%|████████▌ | 109/128 [00:57<00:04,  4.22it/s]
batch:  86%|████████▌ | 110/128 [00:57<00:04,  4.22it/s]
batch:  87%|████████▋ | 111/128 [00:57<00:04,  4.24it/s]
batch:  88%|████████▊ | 112/128 [00:57<00:03,  4.23it/s]
batch:  88%|████████▊ | 113/128 [00:57<00:03,  4.24it/s]
batch:  89%|████████▉ | 114/128 [00:58<00:03,  4.24it/s]
batch:  90%|████████▉ | 115/128 [00:58<00:03,  4.23it/s]
batch:  91%|█████████ | 116/128 [00:58<00:02,  4.25it/s]
batch:  91%|█████████▏| 117/128 [00:58<00:02,  4.25it/s]
batch:  92%|█████████▏| 118/128 [00:59<00:02,  4.26it/s]
batch:  93%|█████████▎| 119/128 [00:59<00:02,  4.26it/s]
batch:  94%|█████████▍| 120/128 [00:59<00:01,  4.23it/s]
batch:  95%|█████████▍| 121/128 [00:59<00:01,  4.22it/s]
batch:  95%|█████████▌| 122/128 [01:00<00:01,  4.20it/s]
batch:  96%|█████████▌| 123/128 [01:00<00:01,  4.19it/s]
batch:  97%|█████████▋| 124/128 [01:00<00:00,  4.19it/s]
batch:  98%|█████████▊| 125/128 [01:00<00:00,  4.19it/s]
batch:  98%|█████████▊| 126/128 [01:01<00:00,  4.20it/s]
batch:  99%|█████████▉| 127/128 [01:01<00:00,  4.20it/s]
batch: 100%|██████████| 128/128 [01:01<00:00,  4.75it/s]
batch: 100%|██████████| 128/128 [01:01<00:00,  2.08it/s]

  0%|          | 0/8160 [00:00<?, ?it/s]
  1%|▏         | 116/8160 [00:00<00:06, 1151.32it/s]
  3%|▎         | 232/8160 [00:00<00:06, 1155.36it/s]
  4%|▍         | 349/8160 [00:00<00:06, 1152.17it/s]
  6%|▌         | 467/8160 [00:00<00:06, 1160.82it/s]
  7%|▋         | 584/8160 [00:00<00:06, 1143.18it/s]
  9%|▊         | 703/8160 [00:00<00:06, 1151.69it/s]
 10%|█         | 819/8160 [00:00<00:06, 1119.50it/s]
 11%|█▏        | 937/8160 [00:00<00:06, 1132.21it/s]
 13%|█▎        | 1055/8160 [00:00<00:06, 1140.93it/s]
 14%|█▍        | 1172/8160 [00:01<00:06, 1149.02it/s]
 16%|█▌        | 1287/8160 [00:01<00:06, 1119.49it/s]
 17%|█▋        | 1402/8160 [00:01<00:05, 1127.76it/s]
 19%|█▊        | 1517/8160 [00:01<00:05, 1132.57it/s]
 20%|█▉        | 1631/8160 [00:01<00:05, 1131.74it/s]
 21%|██▏       | 1745/8160 [00:01<00:05, 1131.41it/s]
 23%|██▎       | 1859/8160 [00:01<00:05, 1132.07it/s]
 24%|██▍       | 1973/8160 [00:01<00:05, 1129.77it/s]
 26%|██▌       | 2087/8160 [00:01<00:05, 1128.01it/s]
 27%|██▋       | 2201/8160 [00:01<00:05, 1128.87it/s]
 28%|██▊       | 2314/8160 [00:02<00:05, 1126.28it/s]
 30%|██▉       | 2427/8160 [00:02<00:05, 1115.89it/s]
 31%|███       | 2539/8160 [00:02<00:05, 1115.72it/s]
 32%|███▏      | 2651/8160 [00:02<00:05, 1083.28it/s]
 34%|███▍      | 2764/8160 [00:02<00:04, 1091.69it/s]
 35%|███▌      | 2876/8160 [00:02<00:04, 1096.32it/s]
 37%|███▋      | 2988/8160 [00:02<00:04, 1101.06it/s]
 38%|███▊      | 3099/8160 [00:02<00:04, 1079.36it/s]
 39%|███▉      | 3211/8160 [00:02<00:04, 1085.05it/s]
 41%|████      | 3324/8160 [00:02<00:04, 1093.67it/s]
 42%|████▏     | 3434/8160 [00:03<00:04, 1085.79it/s]
 43%|████▎     | 3546/8160 [00:03<00:04, 1095.28it/s]
 45%|████▍     | 3656/8160 [00:03<00:04, 1089.85it/s]
 46%|████▌     | 3766/8160 [00:03<00:04, 1089.03it/s]
 47%|████▋     | 3875/8160 [00:03<00:03, 1082.14it/s]
 49%|████▉     | 3984/8160 [00:03<00:03, 1082.35it/s]
 50%|█████     | 4093/8160 [00:03<00:03, 1082.17it/s]
 51%|█████▏    | 4202/8160 [00:03<00:03, 1073.24it/s]
 53%|█████▎    | 4312/8160 [00:03<00:03, 1076.48it/s]
 54%|█████▍    | 4420/8160 [00:03<00:03, 1050.85it/s]
 56%|█████▌    | 4529/8160 [00:04<00:03, 1061.43it/s]
 57%|█████▋    | 4639/8160 [00:04<00:03, 1069.86it/s]
 58%|█████▊    | 4749/8160 [00:04<00:03, 1076.85it/s]
 60%|█████▉    | 4857/8160 [00:04<00:03, 1058.04it/s]
 61%|██████    | 4966/8160 [00:04<00:02, 1065.23it/s]
 62%|██████▏   | 5076/8160 [00:04<00:02, 1069.57it/s]
 64%|██████▎   | 5184/8160 [00:04<00:02, 1059.81it/s]
 65%|██████▍   | 5291/8160 [00:04<00:02, 1059.74it/s]
 66%|██████▌   | 5398/8160 [00:04<00:02, 1056.80it/s]
 67%|██████▋   | 5504/8160 [00:05<00:02, 1054.61it/s]
 69%|██████▉   | 5610/8160 [00:05<00:02, 1053.61it/s]
 70%|███████   | 5716/8160 [00:05<00:02, 1049.81it/s]
 71%|███████▏  | 5821/8160 [00:05<00:02, 1045.09it/s]
 73%|███████▎  | 5926/8160 [00:05<00:02, 1046.03it/s]
 74%|███████▍  | 6031/8160 [00:05<00:02, 1019.17it/s]
 75%|███████▌  | 6139/8160 [00:05<00:01, 1032.76it/s]
 77%|███████▋  | 6247/8160 [00:05<00:01, 1043.37it/s]
 78%|███████▊  | 6354/8160 [00:05<00:01, 1047.34it/s]
 79%|███████▉  | 6459/8160 [00:05<00:01, 1027.72it/s]
 80%|████████  | 6564/8160 [00:06<00:01, 1029.77it/s]
 82%|████████▏ | 6670/8160 [00:06<00:01, 1038.03it/s]
 83%|████████▎ | 6774/8160 [00:06<00:01, 1029.56it/s]
 84%|████████▍ | 6880/8160 [00:06<00:01, 1037.50it/s]
 86%|████████▌ | 6984/8160 [00:06<00:01, 1036.67it/s]
 87%|████████▋ | 7088/8160 [00:06<00:01, 1029.75it/s]
 88%|████████▊ | 7192/8160 [00:06<00:00, 1026.59it/s]
 89%|████████▉ | 7295/8160 [00:06<00:00, 1021.66it/s]
 91%|█████████ | 7398/8160 [00:06<00:00, 1023.77it/s]
 92%|█████████▏| 7501/8160 [00:06<00:00, 1019.70it/s]
 93%|█████████▎| 7603/8160 [00:07<00:00, 1019.70it/s]
 94%|█████████▍| 7705/8160 [00:07<00:00, 995.25it/s] 
 96%|█████████▌| 7810/8160 [00:07<00:00, 1006.50it/s]
 97%|█████████▋| 7914/8160 [00:07<00:00, 1013.58it/s]
 98%|█████████▊| 8018/8160 [00:07<00:00, 1017.43it/s]
100%|█████████▉| 8120/8160 [00:07<00:00, 999.32it/s] 
100%|██████████| 8160/8160 [00:07<00:00, 1072.75it/s]

  0%|          | 0/8160 [00:00<?, ?it/s]
  1%|▏         | 118/8160 [00:00<00:06, 1174.74it/s]
  3%|▎         | 238/8160 [00:00<00:06, 1188.59it/s]
  4%|▍         | 357/8160 [00:00<00:06, 1181.32it/s]
  6%|▌         | 476/8160 [00:00<00:06, 1147.40it/s]
  7%|▋         | 593/8160 [00:00<00:06, 1154.61it/s]
  9%|▊         | 711/8160 [00:00<00:06, 1162.92it/s]
 10%|█         | 830/8160 [00:00<00:06, 1168.09it/s]
 12%|█▏        | 947/8160 [00:00<00:06, 1161.17it/s]
 13%|█▎        | 1064/8160 [00:00<00:06, 1161.75it/s]
 14%|█▍        | 1181/8160 [00:01<00:06, 1158.27it/s]
 16%|█▌        | 1297/8160 [00:01<00:05, 1157.37it/s]
 17%|█▋        | 1415/8160 [00:01<00:05, 1158.60it/s]
 19%|█▉        | 1531/8160 [00:01<00:05, 1156.39it/s]
 20%|██        | 1649/8160 [00:01<00:05, 1160.89it/s]
 22%|██▏       | 1766/8160 [00:01<00:05, 1156.84it/s]
 23%|██▎       | 1882/8160 [00:01<00:05, 1150.85it/s]
 24%|██▍       | 1998/8160 [00:01<00:05, 1146.70it/s]
 26%|██▌       | 2115/8160 [00:01<00:05, 1152.83it/s]
 27%|██▋       | 2231/8160 [00:01<00:05, 1137.56it/s]
 29%|██▉       | 2347/8160 [00:02<00:05, 1139.82it/s]
 30%|███       | 2462/8160 [00:02<00:05, 1136.34it/s]
 32%|███▏      | 2576/8160 [00:02<00:05, 1105.81it/s]
 33%|███▎      | 2690/8160 [00:02<00:04, 1110.03it/s]
 34%|███▍      | 2802/8160 [00:02<00:04, 1112.89it/s]
 36%|███▌      | 2915/8160 [00:02<00:04, 1117.31it/s]
 37%|███▋      | 3028/8160 [00:02<00:04, 1120.75it/s]
 38%|███▊      | 3141/8160 [00:02<00:04, 1094.01it/s]
 40%|███▉      | 3252/8160 [00:02<00:04, 1097.05it/s]
 41%|████      | 3365/8160 [00:02<00:04, 1104.41it/s]
 43%|████▎     | 3477/8160 [00:03<00:04, 1105.53it/s]
 44%|████▍     | 3591/8160 [00:03<00:04, 1110.58it/s]
 45%|████▌     | 3703/8160 [00:03<00:04, 1102.78it/s]
 47%|████▋     | 3815/8160 [00:03<00:03, 1101.89it/s]
 48%|████▊     | 3926/8160 [00:03<00:03, 1103.40it/s]
 49%|████▉     | 4037/8160 [00:03<00:03, 1100.05it/s]
 51%|█████     | 4148/8160 [00:03<00:03, 1098.11it/s]
 52%|█████▏    | 4258/8160 [00:03<00:03, 1097.49it/s]
 54%|█████▎    | 4368/8160 [00:03<00:03, 1095.46it/s]
 55%|█████▍    | 4480/8160 [00:03<00:03, 1096.98it/s]
 56%|█████▋    | 4591/8160 [00:04<00:03, 1100.71it/s]
 58%|█████▊    | 4702/8160 [00:04<00:03, 1097.50it/s]
 59%|█████▉    | 4812/8160 [00:04<00:03, 1096.07it/s]
 60%|██████    | 4923/8160 [00:04<00:02, 1098.03it/s]
 62%|██████▏   | 5033/8160 [00:04<00:02, 1097.39it/s]
 63%|██████▎   | 5143/8160 [00:04<00:02, 1087.37it/s]
 64%|██████▍   | 5252/8160 [00:04<00:02, 1082.39it/s]
 66%|██████▌   | 5361/8160 [00:04<00:02, 1078.07it/s]
 67%|██████▋   | 5470/8160 [00:04<00:02, 1080.98it/s]
 68%|██████▊   | 5579/8160 [00:04<00:02, 1065.23it/s]
 70%|██████▉   | 5688/8160 [00:05<00:02, 1066.39it/s]
 71%|███████   | 5796/8160 [00:05<00:02, 1069.45it/s]
 72%|███████▏  | 5905/8160 [00:05<00:02, 1069.38it/s]
 74%|███████▎  | 6012/8160 [00:05<00:02, 1037.04it/s]
 75%|███████▌  | 6120/8160 [00:05<00:01, 1048.15it/s]
 76%|███████▋  | 6227/8160 [00:05<00:01, 1054.18it/s]
 78%|███████▊  | 6335/8160 [00:05<00:01, 1057.86it/s]
 79%|███████▉  | 6442/8160 [00:05<00:01, 1059.91it/s]
 80%|████████  | 6549/8160 [00:05<00:01, 1058.30it/s]
 82%|████████▏ | 6655/8160 [00:06<00:01, 1033.59it/s]
 83%|████████▎ | 6763/8160 [00:06<00:01, 1041.43it/s]
 84%|████████▍ | 6870/8160 [00:06<00:01, 1046.11it/s]
 85%|████████▌ | 6976/8160 [00:06<00:01, 1046.70it/s]
 87%|████████▋ | 7081/8160 [00:06<00:01, 1045.72it/s]
 88%|████████▊ | 7187/8160 [00:06<00:00, 1049.14it/s]
 89%|████████▉ | 7292/8160 [00:06<00:00, 1033.52it/s]
 91%|█████████ | 7398/8160 [00:06<00:00, 1035.37it/s]
 92%|█████████▏| 7503/8160 [00:06<00:00, 1038.67it/s]
 93%|█████████▎| 7608/8160 [00:06<00:00, 1041.55it/s]
 95%|█████████▍| 7713/8160 [00:07<00:00, 1034.40it/s]
 96%|█████████▌| 7818/8160 [00:07<00:00, 1037.25it/s]
 97%|█████████▋| 7922/8160 [00:07<00:00, 1037.50it/s]
 98%|█████████▊| 8027/8160 [00:07<00:00, 1038.89it/s]
100%|█████████▉| 8131/8160 [00:07<00:00, 1036.01it/s]
100%|██████████| 8160/8160 [00:07<00:00, 1093.83it/s]
2026-01-05 00:22:04.558424: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:22:13.316745: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:22:13.317809: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcuda.so.1
2026-01-05 00:22:13.344368: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-01-05 00:22:13.344436: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:22:13.349302: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:22:13.349391: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:22:13.351914: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:22:13.353569: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:22:13.357036: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:22:13.359313: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:22:13.360787: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:22:13.361173: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:22:13.361470: 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 AVX512F FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2026-01-05 00:22:13.363090: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
2026-01-05 00:22:13.363363: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1720] Found device 0 with properties: 
pciBusID: 0000:61:00.0 name: NVIDIA L40S computeCapability: 8.9
coreClock: 2.52GHz coreCount: 142 deviceMemorySize: 44.52GiB deviceMemoryBandwidth: 804.75GiB/s
2026-01-05 00:22:13.363396: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:22:13.363412: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:22:13.363425: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:22:13.363438: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcufft.so.10
2026-01-05 00:22:13.363451: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcurand.so.10
2026-01-05 00:22:13.363464: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusolver.so.10
2026-01-05 00:22:13.363476: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcusparse.so.11
2026-01-05 00:22:13.363488: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:22:13.363787: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1862] Adding visible gpu devices: 0
2026-01-05 00:22:13.363813: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
2026-01-05 00:22:13.778569: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1261] Device interconnect StreamExecutor with strength 1 edge matrix:
2026-01-05 00:22:13.778665: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1267]      0 
2026-01-05 00:22:13.778675: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1280] 0:   N 
2026-01-05 00:22:13.779372: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1406] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 42313 MB memory) -> physical GPU (device: 0, name: NVIDIA L40S, pci bus id: 0000:61:00.0, compute capability: 8.9)
2026-01-05 00:22:13.819246: 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:22:13.833780: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2500000000 Hz
2026-01-05 00:22:19.909893: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublas.so.11
2026-01-05 00:22:20.432571: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcublasLt.so.11
2026-01-05 00:22:20.437411: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudnn.so.8
2026-01-05 00:22:22.012361: W tensorflow/stream_executor/gpu/asm_compiler.cc:63] Running ptxas --version returned 256
2026-01-05 00:22:22.134577: 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:22:44.043471: 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)