src/net_types.py


1
2
3
4
5
6
7
8
9

import classification
import counting_small
import counting_big

types = {
    'classification': classification,
    'counting-small': counting_small,
    'counting-big': counting_big,
}