Skip to content

Commit eb55d80

Browse files
committed
added citation for resnet
1 parent c52221d commit eb55d80

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)