mirror of
				https://github.com/facebookresearch/sam2.git
				synced 2025-11-04 19:42:12 +08:00 
			
		
		
		
	Update hieradet.py
ufmt formatting fixed.
This commit is contained in:
		
							parent
							
								
									102ddb8899
								
							
						
					
					
						commit
						46945a2122
					
				@ -47,7 +47,6 @@ class MultiScaleAttention(nn.Module):
 | 
			
		||||
        self.dim = dim
 | 
			
		||||
        self.dim_out = dim_out
 | 
			
		||||
        self.num_heads = num_heads
 | 
			
		||||
    
 | 
			
		||||
        self.q_pool = q_pool
 | 
			
		||||
        self.qkv = nn.Linear(dim, dim_out * 3)
 | 
			
		||||
        self.proj = nn.Linear(dim_out, dim_out)
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user