add trained models

.gitignore

@@ -1,19 +1,10 @@
+__pycache__/
 data/
-p/
 pack/
-pack_old/
-__pycache__/
-*random_result/
-_*
-bk/
+pack_net/RL*
+pack_net/DL*
+compare/all_img/
 *.jpg
 *.pdf
-final_bk/
-for-5/
 *.txt
-compare/*.txt
-tmp/
-pack_net/RL*
-pack_net/DL*
-first_version_bk/
 *.pyc

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python.pythonPath": "/opt/anaconda3/bin/python"
+}

scripts/test.sh

@@ -21,4 +21,4 @@ python trainer.py \
     --initial_container_width=7 \
     --initial_container_height=50 \
     \
-    --checkpoint=
+    --checkpoint=./p/2d-bot-C+P+S-lb-soft-width-5-note-sh-R-diff/

trainer.py

@@ -525,18 +525,18 @@ def train_pack(**kwargs):
     # Data settings
     parser.add_argument('--obj_dim', default=2, type=int)
     parser.add_argument('--nodes', dest='num_nodes', default=6, type=int)
-    parser.add_argument('--total_obj_num', default=10, type=int)            # if more, do Rolling @TODO
+    parser.add_argument('--total_obj_num', default=10, type=int)            # if more, do Rolling, todo
     parser.add_argument('--dataset', default='RAND', type=str)              # RAND, PPSG, MIX
     # sizes of blocks and containers
     parser.add_argument('--unit', default=1.0, type=float)
     parser.add_argument('--arm_size', default=1, type=int) # size of robotic arm to pass and rotate a block
     parser.add_argument('--min_size', default=1, type=int)
     parser.add_argument('--max_size', default=5, type=int)
     parser.add_argument('--container_width', default=5, type=int)
-    parser.add_argument('--container_length', default=5, type=int)  # for 3D
+    parser.add_argument('--container_length', default=5, type=int)  # for 3D, todo
     parser.add_argument('--container_height', default=50, type=int)
     parser.add_argument('--initial_container_width', default=7, type=int)
-    parser.add_argument('--initial_container_length', default=7, type=int)  # for 3D
+    parser.add_argument('--initial_container_length', default=7, type=int)  # for 3D, todo
     parser.add_argument('--initial_container_height', default=50, type=int)
 
     # Packing settings