import os
import sys

project_root = '/home/yhsun/classification-main/'
sys.path.append(project_root)
print("Project root added to sys.path:", project_root)

# Verify that we can access the model package directly
import model
print("Model package is accessible, path:", model.__file__)

from model.layer import linear_head
print("Imported linear_head from model.layer")