in dataflux_pytorch/benchmark/checkpointing/multinode/train_async_save.py [0:0]
def __init__( self, vocab_size: int = 33278, nlayers: int = 2, simulated_workload_duration: int = 0,