Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

.from_pretrained for 3d inputs #14

Open
mahootiha-maryam opened this issue Aug 1, 2022 · 2 comments
Open

.from_pretrained for 3d inputs #14

mahootiha-maryam opened this issue Aug 1, 2022 · 2 comments

Comments

@mahootiha-maryam
Copy link

Cant we use pretrained models for 3d images? I had 3d clinical images they are not too much and I wanted to use them for image classification. so I needed a kind of pretrained model. I saw repository but I couldnt understand can I use .from_pretrained for 3d or not. If it is possible can you give example by code?

@mahootiha-maryam
Copy link
Author

class EfficientNet3D(nn.Module):
    """
    An EfficientNet model. Most easily loaded with the .from_name or .from_pretrained methods
    Args:
        blocks_args (list): A list of BlockArgs to construct blocks
        global_params (namedtuple): A set of GlobalParams shared between blocks
    Example:
        model = EfficientNet3D.from_pretrained('efficientnet-b0')

You wrote in class EfficientNet3D but when I use this code
model = EfficientNet3D.from_pretrained('efficientnet-b0')
I face this error
type object 'EfficientNet3D' has no attribute 'from_pretrained'

@aaisha-bah
Copy link

Did you manage to solve it? I am facing the same issue

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants