Fixed minor bug in shape validation in SongUNet#1230

CharlelieLrt merged 1 commit intoNVIDIA:mainNVIDIA/physicsnemo:mainfrom

CharlelieLrt:bugfix-songunet-check-shapeCharlelieLrt/modulus:bugfix-songunet-check-shapeCopy head branch name to clipboard