wandb:   1 of 1 files downloaded.
Selecting training samples
Keeping 390473 intervals


Selecting validation samples
Keeping 21987 intervals


Selecting test samples
Keeping 22595 intervals
Final sizes: train: (390473, 3), val: (21987, 3), test: (22595, 3)
GPU available: True (cuda), used: True
TPU available: False, using: 0 TPU cores
HPU available: False, using: 0 HPUs
/opt/conda/lib/python3.11/site-packages/pytorch_lightning/loggers/wandb.py:397: There is a wandb run already in progress and newly created instances of `WandbLogger` will reuse this run. If this is not desired, call `wandb.finish()` before instantiating `WandbLogger`.
LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0,1,2,3,4,5,6,7]
Validation DataLoader 0: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 43/43 [00:09<00:00,  4.60it/s]
/opt/conda/lib/python3.11/site-packages/torchmetrics/utilities/prints.py:43: UserWarning: The variance of predictions or target is close to zero. This can cause instability in Pearson correlationcoefficient, leading to wrong results. Consider re-scaling the input if possible or computing using alarger dtype (currently using torch.float32).
LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0,1,2,3,4,5,6,7]

  | Name         | Type             | Params | Mode
----------------------------------------------------------
0 | model        | DilatedConvModel | 6.3 M  | train
1 | loss         | MSELoss          | 0      | train
2 | activation   | Identity         | 0      | train
3 | val_metrics  | MetricCollection | 0      | train
4 | test_metrics | MetricCollection | 0      | train
5 | transform    | Identity         | 0      | train
----------------------------------------------------------
6.3 M     Trainable params
0         Non-trainable params
6.3 M     Total params
25.358    Total estimated model params size (MB)
131       Modules in train mode
0         Modules in eval mode
Epoch 14: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 763/763 [07:51<00:00,  1.62it/s, v_num=arkd, train_loss_step=0.551, train_loss_epoch=0.456]
/opt/conda/lib/python3.11/site-packages/torchmetrics/utilities/prints.py:43: UserWarning: The variance of predictions or target is close to zero. This can cause instability in Pearson correlationcoefficient, leading to wrong results. Consider re-scaling the input if possible or computing using alarger dtype (currently using torch.float32).
                                                                                                                                                                                                                                           
`Trainer.fit` stopped: `max_epochs=15` reached.