| { | |
| "version": "2.0", | |
| "metadata": { | |
| "description": "A model architecture for 256^3 brains with 5 decoders, where dilation in each is structured as 16->8->4->2->1", | |
| "framework": "tinygrad", | |
| "input_shape": [1, 256, 256, 256], | |
| "output_classes": 5 | |
| }, | |
| "preprocessing": { | |
| "op": "none", | |
| "params": {} | |
| }, | |
| "forward_pass": [ | |
| {"op": "conv3d", "params": {"in_channels": 1, "out_channels": 15, "kernel_size": 3, "padding": 16, "stride": 1, "dilation": 16, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15, "affine" : false}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15, "affine": false}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 8, "stride": 1, "dilation": 8, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15, "affine" : false}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 4, "stride": 1, "dilation": 4, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15, "affine" : false}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 2, "stride": 1, "dilation": 2, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15, "affine" : false}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 1, "stride": 1, "dilation": 1, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 16, "stride": 1, "dilation": 16, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 8, "stride": 1, "dilation": 8, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 4, "stride": 1, "dilation": 4, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 2, "stride": 1, "dilation": 2, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 1, "stride": 1, "dilation": 1, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 16, "stride": 1, "dilation": 16, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 8, "stride": 1, "dilation": 8, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 4, "stride": 1, "dilation": 4, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 2, "stride": 1, "dilation": 2, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 1, "stride": 1, "dilation": 1, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 16, "stride": 1, "dilation": 16, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 8, "stride": 1, "dilation": 8, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 4, "stride": 1, "dilation": 4, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 2, "stride": 1, "dilation": 2, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 1, "stride": 1, "dilation": 1, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 16, "stride": 1, "dilation": 16, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 8, "stride": 1, "dilation": 8, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 4, "stride": 1, "dilation": 4, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 2, "stride": 1, "dilation": 2, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 15, "kernel_size": 3, "padding": 1, "stride": 1, "dilation": 1, "bias": false}}, | |
| {"op": "group_norm", "params": {"num_groups": 15, "num_channels": 15}}, | |
| {"op": "batch_norm3d", "params": {"num_features": 15}}, | |
| {"op": "gelu", "params": {}}, | |
| {"op": "conv3d", "params": {"in_channels": 15, "out_channels": 2, "kernel_size": 1, "padding": 0, "stride": 1, "dilation": 1, "bias": false}} | |
| ], | |
| "postprocessing": { | |
| "op": "none", | |
| "params": {} | |
| } | |
| } | |