-
Notifications
You must be signed in to change notification settings - Fork 58
Open
Description
@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
Labels
No labels