refactor: reorganize project structure by moving core modules and update import paths in API server
This commit is contained in:
@@ -0,0 +1,3 @@
|
||||
import torch
|
||||
checkpoint = torch.load('cloud_removal_model/cloud_removal_unet_best.pth', map_location='cpu')
|
||||
print(list(checkpoint['model_state_dict'].keys())[:5])
|
||||
Reference in New Issue
Block a user