RobustART.model package

Module contents

RobustART.model.get_model(config)

Parameters

config – a dict which saves the model kwargs

Returns

a pytorch model