Skip to content

[TabArena] TabPFNv2 raises exception when parallel bagged without pre-downloaded weights #160

@Innixma

Description

@Innixma

TabPFNv2 raises an exception when parallel bagged without pre-downloaded weights.

I believe each of the ray processes try to download the TabPFNv2 weights at the same time, leading to an exception. We will need to find a workaround for this, ideally one that doesn't involve the user manually downloading the weights.

One solution would be to add special logic in AbstractModel that applies a pre-bagging step executed before all other steps, which will check for the model weights existing and download them if not present. This can be executed at the very start of the BaggedEnsembleModel.fit call.

Because the weights downloaded is dependent on the model hyperparameters (and potentially the problem type), this has to be a step done post model init, or otherwise as a classmethod with hyperparameters as an input argument.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions