ZB
ZB
Did you add the '--max-wh' arg in the export command? It works for me. The error occurs because you do not have efficient nms which is a tensorRT feature.
非常感谢讲解 @HydrogenSulfate ,现在我有点了解应该怎么写边界条件这部分了。如果我的约束条件中的u也是网络输出,而我希望约束du/dx,是不是能像下面这么写: ``` bc = ppsci.constraint.BoundaryConstraint( {"du/dx": lambda out: out['u'].diff(x)}, {"du/dx": lambda out: h * (out['u'] - u0)}, .... ) ``` 其中h和u0是定义好的常量。
@HydrogenSulfate 不好意思又麻烦了。请帮忙看下我这段代码有什么问题: ```@hydra.main(version_base=None, config_path='./', config_name='dummy_config.yaml') def main(cfg: DictConfig): model = ppsci.arch.MLP(**cfg.MODEL) geom = {"cuboid": ppsci.geometry.Cuboid( (-14.0, -2.0, -1.0), (14.0, 2.0, 1.0)) } train_dataloader_cfg = { "dataset": "IterableNamedArrayDataset", "iters_per_epoch": cfg.TRAIN.iters_per_epoch, }...