TheCamusean commited on
Commit
123dbb5
1 Parent(s): 19b30ef

Loaded Partial Pointcloud model

Browse files
partial_grasp_dif/model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bee255a1c3d15539ea912193b84091412277052b807d6713a5fc881798607299
3
+ size 11422297
partial_grasp_dif/params.json ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "Description" : [ "This experiment trains jointly an SDF model and a SE(3) Grasp Energy" ],
3
+ "exp_log_dir": "multiobject_partial_graspdif",
4
+ "single_object": false,
5
+ "TrainSpecs": {
6
+ "batch_size": 2,
7
+ "num_epochs": 90000,
8
+ "steps_til_summary": 500,
9
+ "iters_til_checkpoint": 1000,
10
+ "epochs_til_checkpoint": 10
11
+ },
12
+ "NetworkArch" : "PointcloudGraspDiffusion",
13
+ "NetworkSpecs" : {
14
+ "feature_encoder": {
15
+ "enc_dim": 132,
16
+ "in_dim": 3,
17
+ "out_dim": 7,
18
+ "dims" : [ 512, 512, 512, 512, 512, 512, 512, 512],
19
+ "dropout" : [0, 1, 2, 3, 4, 5, 6, 7],
20
+ "dropout_prob" : 0.2,
21
+ "norm_layers" : [0, 1, 2, 3, 4, 5, 6, 7],
22
+ "latent_in" : [4],
23
+ "xyz_in_all" : false,
24
+ "use_tanh" : false,
25
+ "latent_dropout" : false,
26
+ "weight_norm" : true
27
+ },
28
+ "encoder": {
29
+ "latent_size": 132,
30
+ "hidden_dim": 512
31
+ },
32
+ "points": {
33
+ "n_points": 30,
34
+ "loc": [0.0, 0.0, 0.5],
35
+ "scale": [0.7, 0.5, 0.7]
36
+ },
37
+ "decoder": {
38
+ "hidden_dim": 512
39
+ }
40
+ },
41
+ "LearningRateSchedule" : [
42
+ {
43
+ "Type" : "Step",
44
+ "Initial" : 0.0005,
45
+ "Interval" : 500,
46
+ "Factor" : 0.5
47
+ },
48
+ {
49
+ "Type" : "Step",
50
+ "Initial" : 0.001,
51
+ "Interval" : 500,
52
+ "Factor" : 0.5
53
+ },
54
+ {
55
+ "Type" : "Step",
56
+ "Initial" : 0.001,
57
+ "Interval" : 500,
58
+ "Factor" : 0.5
59
+ }],
60
+ "Losses": ["sdf_loss", "projected_denoising_loss"]
61
+ }