segmentation / train.log
salmanshahid's picture
Upload folder using huggingface_hub
07d5bbe verified
[2021-03-19 18:29:57,529][lightning][INFO] - GPU available: True, used: True
[2021-03-19 18:29:57,531][lightning][INFO] - TPU available: None, using: 0 TPU cores
[2021-03-19 18:29:57,531][lightning][INFO] - LOCAL_RANK: 0 - CUDA_VISIBLE_DEVICES: [0,1,2,3]
[2021-03-19 18:30:08,622][lightning][INFO] - initializing ddp: GLOBAL_RANK: 0, MEMBER: 1/4
[2021-03-19 18:32:58,993][lightning][INFO] - Set SLURM handle signals.
[2021-03-19 18:32:59,068][lightning][INFO] -
| Name | Type | Params | In sizes | Out sizes
------------------------------------------------------------------------------------------------------------
0 | sincnet | SincNet | 42.6 K | [32, 1, 80000] | [32, 60, 293]
1 | lstm | LSTM | 1.4 M | [32, 293, 60] | [[32, 293, 256], [[8, 32, 128], [8, 32, 128]]]
2 | linear | ModuleList | 49.4 K | ? | ?
3 | classifier | Linear | 516 | [32, 293, 128] | [32, 293, 4]
4 | activation | Sigmoid | 0 | [32, 293, 4] | [32, 293, 4]
------------------------------------------------------------------------------------------------------------
1.5 M Trainable params
0 Non-trainable params
1.5 M Total params
[2021-03-23 02:26:47,615][lightning][INFO] - bypassing sigterm