You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Ross Wightman 16f7aa9f54 Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models		3 years ago
..
__init__.py	Basic forward pass test for all registered models	4 years ago
test_layers.py	Add custom grad tests, fix cut & paste error with hard_mish ME, add a few more pytorch act fns to factory	4 years ago
test_models.py	Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models	3 years ago