4 lines
163 B
Python
4 lines
163 B
Python
import torch
|
|
checkpoint = torch.load('cloud_removal_model/cloud_removal_unet_best.pth', map_location='cpu')
|
|
print(list(checkpoint['model_state_dict'].keys())[:5])
|