add train

4d9ae840 · 朱学凯 · 44a442cd · 690c8f4d · 4d9ae840 · 4d9ae840
--- a/.ipynb_checkpoints/pre_test-checkpoint.sh
+++ b/.ipynb_checkpoints/pre_test-checkpoint.sh
-CUDA_VISIBLE_DEVICES=1 python run_interaction.py --task=test --b=64 --output=./predict/lr-1e-5-batch-32-e-10-layer6-0420-step-74133 --config=./config/config_layer_6.json --init=./model/lr-1e-5-batch-32-e-10-layer6-0420/epoch-8-step-74133-loss-0.8989318833651185.pth --do_eval=True
\ No newline at end of file
+CUDA_VISIBLE_DEVICES=7 python run_interaction.py --task=test --b=64 --output=./predict/lr-1e-5-batch-32-e-10-layer6-0428-step-57659 --config=./config/config_layer_6.json --init=./model/lr-1e-5-batch-32-e-10-layer6-0428/epoch-6-step-57659-loss-0.6694820924709807.pth --do_eval=True
\ No newline at end of file
--- a/.ipynb_checkpoints/train-checkpoint.sh
+++ b/.ipynb_checkpoints/train-checkpoint.sh
-CUDA_VISIBLE_DEVICES=4 python run_interaction.py --b=32 --task=train --epochs=10 --lr=1e-5 --savedir=lr-1e-5-batch-32-e-10-layer9-0420 --config=./config/config_layer_9.json
\ No newline at end of file
+CUDA_VISIBLE_DEVICES=1 python run_interaction.py --b=32 --task=train --epochs=10 --lr=1e-5 --savedir=lr-1e-5-batch-32-e-10-layer6-0428 --config=./config/config_layer_6.json
\ No newline at end of file
--- a/b-32-layer6-e10.txt
+++ b/b-32-layer6-e10.txt
--- a/b-32-layer9-e10.txt
+++ b/b-32-layer9-e10.txt
--- a/log/lr-1e-5-batch-32-e-10-layer6-0428/events.out.tfevents.1619578126.b1393040f57d.443.0
+++ b/log/lr-1e-5-batch-32-e-10-layer6-0428/events.out.tfevents.1619578126.b1393040f57d.443.0
--- a/log/lr-1e-5-batch-32-e-10-layer9-0428/events.out.tfevents.1619577947.b1393040f57d.336.0
+++ b/log/lr-1e-5-batch-32-e-10-layer9-0428/events.out.tfevents.1619577947.b1393040f57d.336.0
--- a/pre_test.sh
+++ b/pre_test.sh
-CUDA_VISIBLE_DEVICES=1 python run_interaction.py --task=test --b=64 --output=./predict/lr-1e-5-batch-32-e-10-layer6-0420-step-74133 --config=./config/config_layer_6.json --init=./model/lr-1e-5-batch-32-e-10-layer6-0420/epoch-8-step-74133-loss-0.8989318833651185.pth --do_eval=True
\ No newline at end of file
+CUDA_VISIBLE_DEVICES=7 python run_interaction.py --task=test --b=64 --output=./predict/lr-1e-5-batch-32-e-10-layer6-0428-step-57659 --config=./config/config_layer_6.json --init=./model/lr-1e-5-batch-32-e-10-layer6-0428/epoch-6-step-57659-loss-0.6694820924709807.pth --do_eval=True
\ No newline at end of file
--- a/predict/lr-1e-5-batch-32-e-10-layer6-0428-step-57659/eval_results
+++ b/predict/lr-1e-5-batch-32-e-10-layer6-0428-step-57659/eval_results
+RMSE : 1.0396133662472935 ; Pearson Correlation Coefficient : 0.7084505325268997
\ No newline at end of file
--- a/predict/lr-1e-5-batch-32-e-10-layer6-0428-step-57659/test.txt
+++ b/predict/lr-1e-5-batch-32-e-10-layer6-0428-step-57659/test.txt
--- a/train.sh
+++ b/train.sh
+<<<<<<< HEAD
 CUDA_VISIBLE_DEVICES=4 python run_interaction.py --b=32 --task=train_z --epochs=10 --lr=1e-5 --savedir=lr-1e-5-batch-32-e-10-layer9-0420 --config=./config/config_layer_9.json
+=======
+CUDA_VISIBLE_DEVICES=1 python run_interaction.py --b=32 --task=train --epochs=10 --lr=1e-5 --savedir=lr-1e-5-batch-32-e-10-layer6-0428 --config=./config/config_layer_6.json
+>>>>>>> 690c8f4d13462c41d999e0b43572bb005ba22b8e