mmselfsup/configs/benchmarks/classification/places205/resnet50-sobel_mhead_8xb32-steplr-28e_places205.py
2021-12-15 19:09:54 +08:00

5 lines
134 B
Python

_base_ = 'resnet50_mhead_8xb32-steplr-28e_places205.py'
# model settings
model = dict(with_sobel=True, backbone=dict(in_channels=2))