3 lines
94 B
Python
3 lines
94 B
Python
|
model = dict(type='CustomModel', in_channels=[1, 2, 3])
|
||
|
|
optimizer = dict(type='SGD', lr=0.01)
|