Resolve training loss spike

[ssenan]

• After splitting up single script version of code into separate files, there has been an issue where loss appears to randomly spike and breaks learning
• Likely will require a significant rewrite of src/dnadiffusion to isolate how error was introduced