Commit 507c1dfc authored by Ferreira Da Silva, Rafael's avatar Ferreira Da Silva, Rafael
Browse files

Merge branch 'master' of https://code.ornl.gov/workflow/olcf-6-benchmark

parents ced91132 2d470338

env.sh

+1 −1
Original line number Diff line number Diff line 
module load PrgEnv-gnu

module load gcc/10.3.0

module load rocm/6.0.0

ROOT=/lustre/orion/world-shared/stf053/olcf-6-benchmark/miniconda3

ROOT=REPLACE_PWD/miniconda3

export PATH=${ROOT}/bin:$PATH

source ${ROOT}/etc/profile.d/conda.sh

conda activate ${ROOT}/../topaz_env

train.py

+1 −1
Original line number Diff line number Diff line
@@ -253,7 +253,7 @@ if __name__ == "__main__": 
    subsocket.setsockopt_string(zmq.SUBSCRIBE, "metadata")

    subsocket.setsockopt_string(zmq.SUBSCRIBE, "data")

    subsocket.setsockopt_string(zmq.SUBSCRIBE, "END")

    subsocket.connect("tcp://login10.frontier.olcf.ornl.gov:5560")

    subsocket.connect("tcp://REPLACE_HOST:5560")



    print(f"Rank-{torch.distributed.get_rank()}: Synchronized; ready to receive data")

    start_time = datetime.now()