OneScience-Group/Aardvark-Weather/official-src/aardvark main
Imports Python's pickle serialization modulePython file opened for write or append
SHA-256a43c16bc831735ae1ca0d43946abf1da975ecb049e6364cac16d75082a5e4967
MaleculeH₂(Db₂F)
Evidence
1import os
2import sys
3import pickle
4import argparse
5
⋯4 lines
20:18… ns import WeightedRmseLoss, PressureWeightedRmseLoss, RmseLoss
21
22torch.set_float32_matmul_precision("medium")
23
24
⋯10 lines
⋯4 lines
236
237 # Save config
238 with open(output_dir + "/config.pkl", "wb") as f:
239 pickle.dump(vars(args), f)
240
241 world_size = torch.cuda.device_count()
242 mp.spawn(main, args=[world_size, outpu …