| gpu: | |
| device: cpu | |
| infer: | |
| checkpoints: ./checkpoints/segmentation.ckpt | |
| simclr_checkpoint: ./checkpoints/simclr_vitb.ckpt | |
| threshold: 0.5 | |
| image_size: [96, 96, 96] | |
| model: | |
| img_size: [96, 96, 96] | |
| in_channels: 1 | |
| out_channels: 1 | |
| training: | |
| sw_batch_size: 2 |
| gpu: | |
| device: cpu | |
| infer: | |
| checkpoints: ./checkpoints/segmentation.ckpt | |
| simclr_checkpoint: ./checkpoints/simclr_vitb.ckpt | |
| threshold: 0.5 | |
| image_size: [96, 96, 96] | |
| model: | |
| img_size: [96, 96, 96] | |
| in_channels: 1 | |
| out_channels: 1 | |
| training: | |
| sw_batch_size: 2 |