| 
						
						
							
								
							
						
						
					 | 
					@ -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( | 
				
			
			
		
	
	
		
		
			
				
					| 
						
							
								
							
						
						
						
					 | 
					
  |