mmselfsup/configs/benchmarks/classification/imagenet/resnet50-sobel_mhead_linear-8xb32-steplr-90e_in1k.py

_base_ = 'resnet50_mhead_linear-8xb32-steplr-90e_in1k.py'

# model settings
model = dict(with_sobel=True, backbone=dict(in_channels=2, frozen_stages=4))