Browse Source

回滚

master
liupengfei 1 month ago
parent
commit
0038512504
2 changed files with 3 additions and 2 deletions
  1. +1
    -1
      src/model_utils/moxing_adapter.py
  2. +2
    -1
      src/models.py

+ 1
- 1
src/model_utils/moxing_adapter.py View File

@@ -18,7 +18,7 @@
import os
import functools
from mindspore import context
from config import config
from .config import config

_global_sync_count = 0



+ 2
- 1
src/models.py View File

@@ -119,7 +119,8 @@ class FlowNet2(nn.Cell):

# flownetsd
flownetsd_flow2 = self.flownets_d(x)[0]
up3in = flownetsd_flow2 / self.div_flow
# up3in = flownetsd_flow2 / self.div_flow
up3in = flownetsd_flow2
flownetsd_flow = self.upsample3(up3in)
norm_flownetsd_flow = self.channelnorm(flownetsd_flow)



Loading…
Cancel
Save