Proxy: torch.nn.Parameter subclass

[ZoroZhao]

Use torch.nn.Parameter subclass to proxy the model parameter and trace the changes. It is compatible with the torch library.

[Copilot]

Pull Request Overview

This PR introduces a new ProxyParameter subclass of torch.nn.Parameter to enable parameter tracing that is compatible with the PyTorch library. The implementation provides an alternative to the existing proxy wrapper approach by leveraging PyTorch's subclassing mechanism.

Key changes:

• Creates a ProxyParameter class that extends torch.nn.Parameter with tracing capabilities
• Adds support for a new "proxyparameter" model tracker style alongside existing "proxy" and "sampler" modes
• Updates observer and tracer components to handle ProxyParameter instances

Reviewed Changes

Copilot reviewed 8 out of 8 changed files in this pull request and generated 6 comments.

Show a summary per file

File	Description
traincheck/proxy_wrapper/subclass.py	New file implementing ProxyParameter class and proxy_parameter function for parameter tracing
traincheck/proxy_wrapper/proxy_observer.py	Updates observer to handle ProxyParameter instances alongside existing Proxy objects
traincheck/proxy_wrapper/proxy_config.py	Adds proxy_attribute list for filtering ProxyParameter-specific attributes
traincheck/proxy_wrapper/proxy_basics.py	Adds is_proxyparamtetr function to detect ProxyParameter instances
traincheck/instrumentor/tracer.py	Updates tracer to recognize and handle ProxyParameter objects in function arguments
traincheck/instrumentor/source_file.py	Adds "proxyparameter" mode support to model assignment instrumentation
traincheck/instrumentor/dumper.py	Filters out proxy-specific attributes when dumping ProxyParameter objects
traincheck/collect_trace.py	Adds "proxyparameter" as a valid choice for model tracker style

---

_{Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.}

[Copilot]

The logic flow is incorrect. If from_call is True on line 109, phase is set to "call", but then if from_iter is also True, it gets overwritten to "iter". The else clause on line 115 should be elif to properly handle the case where neither from_call nor from_iter is True.

Suggested change

	if from_iter:
	phase = "iter"
	# if the object is generated from getattr, then do not dump it
	elif from_iter:
	phase = "iter"

[Copilot]

Error message is outdated and doesn't include the new "proxyparameter" mode. Should be "Must be one of ['proxy', 'sampler', 'proxyparameter']".

Suggested change

	raise ValueError(f"Invalid mode: {mode}. Must be one of ['proxy', 'sampler']")
	raise ValueError(f"Invalid mode: {mode}. Must be one of ['proxy', 'sampler', 'proxyparameter']")