Skip to content

Minor issue at efficientnet/utils.py  #26

@sharadchandakacherla

Description

@sharadchandakacherla

@yoctta
we get
RuntimeError: expected scalar type Byte but found Float
in the Conv2dStaticSamePadding class' forward method w.r.t self.weights which are of type Parameter

please change the return statement in data/dataset.py getitem method's to

(self.trans(image=image)['image']).float()

to make it work.

I did this in an attempt to make it work, it worked as well, however if you have a better solution please fix it accordingly

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions