Minor fixes
Changed initialization using torch.nn.init
Admins will be upgrading ORNL GitLab Servers on Monday, 13 May 2024, from 5 PM until 7 PM EST. Repositories will experience intermittent outages during this time.
Changed initialization using torch.nn.init