Skip to content

Commit

Permalink
update config
Browse files Browse the repository at this point in the history
  • Loading branch information
mr-superonion committed Sep 17, 2023
1 parent ca87043 commit b77f261
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
2 changes: 1 addition & 1 deletion bin/fpfs_config
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ export TF_CPP_MIN_LOG_LEVEL=10
export XLA_FLAGS="--xla_cpu_multi_thread_eigen=false intra_op_parallelism_threads=1"
export OPENBLAS_NUM_THREADS=1
export MKL_NUM_THREADS=1
export OMP_NUM_THREAD=1
export OMP_NUM_THREADS=1
export XLA_PYTHON_CLIENT_PREALLOCATE="false"
export XLA_PYTHON_CLIENT_ALLOCATOR="platform"

Expand Down
3 changes: 1 addition & 2 deletions bin/fpfs_process_descsim.py
Original file line number Diff line number Diff line change
Expand Up @@ -278,15 +278,14 @@ def run(self, fname):
args = parser.parse_args()
pool = schwimmbad.choose_pool(mpi=args.mpi, processes=args.n_cores)
worker = Worker(args.config)
band = "a"
fname_list = get_sim_fname(
worker.imgdir,
"image",
args.min_id,
args.max_id,
2,
2,
band,
worker.band,
)
for r in pool.map(worker.run, fname_list):
pass
Expand Down

0 comments on commit b77f261

Please sign in to comment.