Open
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
This pull request introduces several improvements to ensure compatibility with recent versions of Python, PyTorch, and scikit-learn, and enhances device management and data loading throughout the AttentionXML codebase. The most significant changes include adding
allow_pickle=Trueto all relevantnp.loadcalls, updating PyTorch and scikit-learn API usage for backward and forward compatibility, and improving device handling for both CPU and GPU environments.Compatibility and Device Management Updates:
allow_pickle=Trueto allnp.loadcalls across the codebase to support loading objects saved with pickling, preventing errors with newer numpy versions. [1] [2] [3] [4] [5] [6] [7] [8] [9] [10]API and Backward Compatibility Fixes:
alphaandvaluekeyword arguments in tensor operations, and ensured correct handling of.item()for scalar tensors. [1] [2] [3]MultiLabelBinarizerusage to handle changes in module paths and initialization, improving backward compatibility. [1] [2]Model and Data Pipeline Improvements:
modules.pyfor better compatibility with PyTorch 2.x and improved padding/masking logic. [1] [2] [3] [4]Documentation and Installation:
README.mdwith a new installation section, including instructions for installing the latest version of PyTorch and using arequirements.txtfile, replacing the old requirements list.These changes collectively modernize the codebase, improve its robustness across environments, and make it easier to set up and run.