Browse Source

fix :trian

final
mckay 7 months ago
parent
commit
66f214790c
  1. 7
      brep2sdf/train.py

7
brep2sdf/train.py

@ -83,12 +83,7 @@ class Trainer:
) )
# 初始化模型 # 初始化模型
self.model = BRepToSDF( self.model = BRepToSDF(config).to(self.device)
brep_feature_dim=config.model.brep_feature_dim,
use_cf=config.model.use_cf,
embed_dim=config.model.embed_dim,
latent_dim=config.model.latent_dim
).to(self.device)
# 初始化优化器 # 初始化优化器
self.optimizer = optim.AdamW( self.optimizer = optim.AdamW(

Loading…
Cancel
Save