@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
@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