Skip to content
Snippets Groups Projects
Commit 1a886efa authored by Anjith GEORGE's avatar Anjith GEORGE
Browse files

Modified the data_folder to be more generic

parent 43fadd2a
Branches
Tags
Loading
Checking pipeline status
Loading
  • Author Maintainer

    @onikisins : This should work for both the cases; I am not sure torch.Tensor(img_array_transformed).unsqueeze(0) is needed here. Can you check if this is ok with you.

  • Contributor

    Thanks @ageorge, I will try to run an experiment with this modification, to make sure there are no side-effects. Will let you know once done.

0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment