None of the images that I got printed are actually that bad size (768x768)
train:   0%|                                                                 | 0/63647 [00:00<?, ?it/s]
Converting from (206, 206, 3)
(206, 206, 4)
Converting from (250, 251, 3)
(250, 251, 4)
torch.Size([1, 4, 768, 768])
['78703.tiff']
batch index 0
Converting from (191, 191, 3)
(191, 191, 4)
train:   0%| | 1/63647 [00:07<135:26:18,  7.66s/it, loss_mask: 1.0000, loss: 1.0000, mask_micro_iou: 0.
torch.Size([1, 4, 768, 768])
['28456.tiff']
batch index 1
Converting from (228, 228, 3)
(228, 228, 4)
train:   0%| | 2/63647 [00:07<59:14:39,  3.35s/it, loss_mask: 1.0000, loss: 1.0000, mask_micro_iou: 0.0
torch.Size([1, 3, 768, 768])
['79771.tiff']
batch index 2
train:   0%| | 2/63647 [00:08<78:48:11,  4.46s/it, loss_mask: 1.0000, loss: 1.0000, mask_micro_iou: 0.0
Traceback (most recent call last):
  File "/usr/lib/python3.6/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/usr/lib/python3.6/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/home/bhaktatejas922/internal-geometry-ml/src/train.py", line 188, in <module>
    main(cfg)
  File "/home/bhaktatejas922/internal-geometry-ml/src/train.py", line 176, in main
    **cfg.training.fit,
  File "/home/bhaktatejas922/internal-geometry-ml/src/training/runner.py", line 268, in fit
    output = self._feed_batch(batch)
  File "/home/bhaktatejas922/internal-geometry-ml/src/training/runner.py", line 35, in wrapped
    res = f(*args, **kwargs)
  File "/home/bhaktatejas922/internal-geometry-ml/src/training/runner.py", line 153, in _feed_batch
    output = self.model(*input)
  File "/home/bhaktatejas922/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/bhaktatejas922/.local/lib/python3.6/site-packages/segmentation_models_pytorch/base/model.py", line 15, in forward
    features = self.encoder(x)
  File "/home/bhaktatejas922/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/bhaktatejas922/.local/lib/python3.6/site-packages/segmentation_models_pytorch/encoders/efficientnet.py", line 50, in forward
    x = self._swish(self._bn0(self._conv_stem(x)))
  File "/home/bhaktatejas922/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/bhaktatejas922/.local/lib/python3.6/site-packages/efficientnet_pytorch/utils.py", line 271, in forward
    x = F.conv2d(x, self.weight, self.bias, self.stride, self.padding, self.dilation, self.groups)
RuntimeError: Given groups=1, weight of size 48 4 3 3, expected input[1, 3, 770, 770] to have 4 channels, but got 3 channels instead