Commit 0fab8927 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add comments about hbutils.random

parent 5f229039

zoo/monochrome/train_.py

+3 −0
Original line number Diff line number Diff line
@@ -98,6 +98,9 @@ def train(dataset_dir: str, session_name: Optional[str] = None, from_ckpt: Optio 
          num_workers: Optional[int] = 8, save_per_epoch: int = 10, eval_epoch: int = 5,

          model_name: str = 'alexnet', seed: Optional[int] = 0):

    if seed is not None:

        # native random, numpy, torch and faker's seeds are includes

        # if you need to register more library for seeding, see:

        # https://hansbug.github.io/hbutils/main/api_doc/random/state.html#register-random-source

        global_seed(seed)



    accelerator = Accelerator(