mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 21:54:47 +08:00
[Fix] Fix a bug that may cause inplace operation error (#884)
* loss * fix * bug
This commit is contained in:
parent
6512f0a0a4
commit
7c9093684e
@ -148,10 +148,10 @@ class FPEM_FFM(BaseModule):
|
|||||||
c4_ffm = c4
|
c4_ffm = c4
|
||||||
c5_ffm = c5
|
c5_ffm = c5
|
||||||
else:
|
else:
|
||||||
c2_ffm += c2
|
c2_ffm = c2_ffm + c2
|
||||||
c3_ffm += c3
|
c3_ffm = c3_ffm + c3
|
||||||
c4_ffm += c4
|
c4_ffm = c4_ffm + c4
|
||||||
c5_ffm += c5
|
c5_ffm = c5_ffm + c5
|
||||||
|
|
||||||
# FFM
|
# FFM
|
||||||
c5 = F.interpolate(
|
c5 = F.interpolate(
|
||||||
|
Loading…
x
Reference in New Issue
Block a user