mmselfsup/configs/benchmarks/classification/places205/resnet50-sobel_mhead_8xb32-steplr-28e_places205.py

5 lines
134 B
Python
Raw Normal View History

2021-12-15 19:09:54 +08:00
_base_ = 'resnet50_mhead_8xb32-steplr-28e_places205.py'
# model settings
model = dict(with_sobel=True, backbone=dict(in_channels=2))