add args for epoch

Author: wangguan1995

examples/euler_beam/euler_beam.py

@@ -28,7 +28,7 @@
     ppsci.utils.misc.set_random_seed(42)
     # set training hyper-parameters
     ITERS_PER_EPOCH = 1
-    EPOCHS = 20 if not args.epochs else args.epochs
+    EPOCHS = 10000 if not args.epochs else args.epochs
     # set output directory
     OUTPUT_DIR = "./output/euler_beam" if not args.output_dir else args.output_dir
     # initialize logger

requirements.txt

@@ -8,7 +8,6 @@ wget
 scipy
 visualdl
 pyvista==0.37.0
-pybind11==2.9.2
 pysdf
 pyyaml
 scikit-optimize

test_tipc/configs/train_eular_beam/train_eular_beam.txt

@@ -4,7 +4,7 @@ python:python3.10
 gpu_list:0
 Global.use_gpu:null
 null:null
-null:null
+--epochs
 null:null
 null:null
 null:null

test_tipc/prepare.sh

@@ -23,4 +23,4 @@ ${pip} install -r requirements.txt
 
 if [ ! -n ${download_dataset}] ; then
     ${python} ${PDSC_DIR}${download_dataset}
-fi
+fi