refactor: adapt Stagate
to BasePretrain
class; improve consistency#204
Merged
RemyLau merged 5 commits intomain from update-spatialFeb 20, 2023
+85-114
Stagate
to BasePretrain
class; improve consistency#204