| import torch | |
| import sys | |
| def inspect(file_path): | |
| print(f"--- Inspecting {file_path} ---") | |
| data = torch.load(file_path, map_location="cpu") | |
| print("Config:", data.get('config', 'No config')) | |
| state_dict = data.get('model_state_dict', {}) | |
| if state_dict: | |
| for k, v in list(state_dict.items())[:10]: | |
| print(f"Layer Key: {k} | Shape: {v.shape}") | |
| print(f"... and {len(state_dict) - 10} more layer keys.") | |
| else: | |
| print("No model_state_dict found.") | |
| if __name__ == "__main__": | |
| inspect("/mnt/e/agent/agent_ai/sail/sail_7c3dcc01.pt") | |