Hi,
I try to replace softmax with binary-representation, repo is here;

  
      
      GitHub
  
  
    

IAMAl/MCWOOHE 1

Experiment to replace Softmax with Integer-Representation - IAMAl/MCWOOHE


  
  
    
    
  
  


Now it works for training, but in a way, it makes an error of
RuntimeError: shape '[64, 1, 324]' is invalid for input of size 10368

Why the input size was changed? although the input is “batch * image_size”.
Best,
S.Takano