Create environment:

conda env create -f environment.yaml
conda activate meshnet
pip install -r requirements.txt

Generate volume meshes:

python meshgen.py step --input meshes/Custom/step --output meshes/Custom/msh --size 0.005  --element-order 1
python meshgen.py step --input meshes/Custom/step --output meshes/Custom/msh --size 0.0025  --element-order 2

Generate datasets:

python data.py meshes/primitives/msh --num_samples 100
python data.py meshes/factory/msh/HexNut2_cg1.msh --num_samples 100

Train:

python train.py --dataset Cuboid200 --num-epochs 50 --learning-rate 1e-4 --batch-size 64 --tensorboard --layers 10
python train.py --dataset Cuboid \
 --num-epochs 500 \
 --learning-rate 1e-4 \
 --batch-size 64 \
 --tensorboard \
 --weighted-loss \
 --alpha 20 \
 --target stress \

Play:

python play.py --checkpoint Cuboid_all_uw --dataset Cuboid1_100 --plots -n 5