17 lines
326 B
Python
17 lines
326 B
Python
|
import os
|
||
|
|
import torch
|
||
|
|
import fire
|
||
|
|
|
||
|
|
|
||
|
|
def printit(p):
|
||
|
|
print(f"printin' in path: {p}")
|
||
|
|
size_initial = os.path.getsize(p)
|
||
|
|
nsd = dict()
|
||
|
|
sd = torch.load(p, map_location="cpu")
|
||
|
|
if "global_step" in sd:
|
||
|
|
print(f"This is global step {sd['global_step']}.")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
fire.Fire(printit)
|