python - 如何导入“Dense3D”并在代码中使用它？

Question

我尝试从 git 运行代码，但发现导入 Dense3D 时出现问题。这里的代码女巫部分与问题有关：

这里是与问题相关的代码部分：

   import torch
   from torch.models import Dense3D
   import torch.nn as nn
   #some lines deleted
   print("Loading options...")
   with open(sys.argv[1], 'r') as optionsFile:
       options = toml.loads(optionsFile.read())
   #Create the model.
   model = Dense3D(options)

它会在另一个尝试中出现“无法导入 Dense3D”或类似的错误，并进行一些更改。

我用 colab 。

标签： pythonpytorchtorch