Results 1 issues of secutron

Thank you for sharing this wonderful works! I found a simple lightning dependancy and epoch parameter in the _cosine_scheduler_fn(...) in Marlin>model>marlin.py cause me an error. Somehow the epoch parameter above...