ShaNet / test.py

Upload 8 files (#1)

336661a verified about 1 year ago

214 Bytes

import torch

try:
    ckpt = torch.load("out/ckpt.pt", map_location="cpu")
    print("✅ Checkpoint has been loaded successfully")
except Exception as e:
    print("❌ Failed to load the checkpoint:", e)

	import torch

	try:
	ckpt = torch.load("out/ckpt.pt", map_location="cpu")
	print("✅ Checkpoint has been loaded successfully")
	except Exception as e:
	print("❌ Failed to load the checkpoint:", e)