lossfile_correction

main_sphere_compression.py

@@ -556,6 +556,7 @@ def main():
         assert validation_dataset.resolution == args.healpix_res, "resolution of validation dataset doesn't match with input dataset"
     # save loss
     if args.loss_file:
+        os.makedirs(args.out_dir, exist_ok=True)
         with open(os.path.join(args.out_dir, args.loss_file), 'a') as f:
             f.write('='*40+f' {datetime.now().strftime("%d-%m-%Y %H:%M:%S")} '+'='*40+'\n')
     # to visualize gradient status