Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Commit 
							
							·
						
						4e2b782
	
1
								Parent(s):
							
							c19a5d3
								
Fix model loading issue
Browse files
    	
        app.py
    CHANGED
    
    | @@ -28,7 +28,6 @@ DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu' | |
| 28 | 
             
            model = NewCRFDepth(version='large07', inv_depth=False,
         | 
| 29 | 
             
                                max_depth=10).to(DEVICE).eval()
         | 
| 30 | 
             
            model = torch.nn.DataParallel(model)
         | 
| 31 | 
            -
            model.load_state_dict()
         | 
| 32 | 
             
            checkpoint = torch.load('checkpoints/nyu_L.pth',
         | 
| 33 | 
             
                                    map_location=torch.device(DEVICE))
         | 
| 34 | 
             
            model.load_state_dict(checkpoint['model'])
         | 
|  | |
| 28 | 
             
            model = NewCRFDepth(version='large07', inv_depth=False,
         | 
| 29 | 
             
                                max_depth=10).to(DEVICE).eval()
         | 
| 30 | 
             
            model = torch.nn.DataParallel(model)
         | 
|  | |
| 31 | 
             
            checkpoint = torch.load('checkpoints/nyu_L.pth',
         | 
| 32 | 
             
                                    map_location=torch.device(DEVICE))
         | 
| 33 | 
             
            model.load_state_dict(checkpoint['model'])
         | 
