diff --git a/bayesflow/experimental/resnet/resnet.py b/bayesflow/experimental/resnet/resnet.py index 862e0ac98..d92cef30a 100644 --- a/bayesflow/experimental/resnet/resnet.py +++ b/bayesflow/experimental/resnet/resnet.py @@ -12,10 +12,14 @@ @serializable("bayesflow.networks", disable_module_check=True) class ResNet(keras.Sequential): """ - Implements the ResNet architecture. + Implements the ResNet architecture, from [1]. Note that we still apply dropout and activation to the output and do not flatten it, so you will need to flatten it yourself and apply at least one linear layer after this network. + + [1] He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep residual learning for image recognition. + In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 770-778). + arXiv:1512.03385 """ def __init__(