Back to home page

EIC code displayed by LXR

 
 

    


File indexing completed on 2025-10-13 08:28:38

0001 #!/bin/bash
0002 ## To be run with 3 arguments defining the run name and the study name for wandb, and gpu_id
0003 
0004 source /cvmfs/sft.cern.ch/lcg/views/LCG_105_cuda/x86_64-el9-gcc11-opt/setup.sh
0005 
0006 nvidia-smi
0007 
0008 pip install numpy h5py matplotlib scipy scikit-learn wandb tf2onnx onnxruntime
0009 
0010 ## Provide your wandb api key here
0011 export WANDB_API_KEY=<WANDB-API-KEY>
0012 
0013 mkdir -p validation checkpoint conversion generation
0014 python /afs/cern.ch/user/p/praikwar/public/par04/training/train.py --run-name $1 --study-name $2 --gpu-ids $3
0015