Skip to content

Commit 10db509

Browse files
eodolevpratz
andauthored
Add citation for resnet (#537) [no ci]
* added citation for resnet * minor formatting --------- Co-authored-by: Valentin Pratz <git@valentinpratz.de>
1 parent ebbddce commit 10db509

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

bayesflow/experimental/resnet/resnet.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,14 @@
1212
@serializable("bayesflow.networks", disable_module_check=True)
1313
class ResNet(keras.Sequential):
1414
"""
15-
Implements the ResNet architecture.
15+
Implements the ResNet architecture, from [1].
1616
1717
Note that we still apply dropout and activation to the output and do not flatten it,
1818
so you will need to flatten it yourself and apply at least one linear layer after this network.
19+
20+
[1] He, K., Zhang, X., Ren, S., & Sun, J. (2016). Deep residual learning for image recognition.
21+
In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 770-778).
22+
arXiv:1512.03385
1923
"""
2024

2125
def __init__(

0 commit comments

Comments
 (0)