Skip to content

Model not working forking from the master branch #2

@yuzhe0690

Description

@yuzhe0690
  1. Several parameters of multiple functions were missing in the master branch's train.py, these parameters existed in the development branch's train.py.
  2. The CE loss within the master branch is causing an error where target 255 is out of bounds. The min-max normalizing in the dataset.py is missing. Once the normalization code is added back, the model is working.

Please include lines of codes that were supposed to be in the file. I spent one whole week trying to fix this on my own as a student trying to use this model for their final year project.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions