Add gasa_generalist (renamed gasa_E_box_50s)

checkpoints/gasa_generalist/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:14f5779978fd92506a77a09c76927eaaa15c0450d1c14d26a6cea541dd169720
+size 170785061

checkpoints/gasa_generalist/config.json

@@ -0,0 +1,201 @@
+{
+  "sam3_defaults": false,
+  "da3_model": "depth-anything/DA3NESTED-GIANT-LARGE",
+  "da3_resolution": null,
+  "use_cached_depth": true,
+  "use_cached_pi3x": false,
+  "pi3x_cache_name": "pi3xvo_cache",
+  "da3_cache_name": "da3_nested_cache_1008",
+  "d_model": 256,
+  "n_heads": 8,
+  "num_decoder_layers": 6,
+  "num_queries": 10,
+  "multi_object": false,
+  "match_strategy": "hungarian",
+  "per_text_decode": false,
+  "grouped_text_attn": false,
+  "sam3_multi_object": false,
+  "num_objects": 1,
+  "max_objects": 0,
+  "train_seghead": false,
+  "train_pixel_decoder": true,
+  "dataset": "scannetpp",
+  "frames_per_sequence": 50,
+  "samples_per_sequence": 1,
+  "part_query_mode": "random",
+  "augment": false,
+  "data_root": null,
+  "split": "train_balanced_v4",
+  "views": 8,
+  "sampling_strategy": "stratified",
+  "da3_chunk_size": 8,
+  "max_scenes": 50,
+  "resolution": 504,
+  "mask_size": 504,
+  "min_mask_coverage": 0.001,
+  "epochs": 20,
+  "stop_at_epoch": 0,
+  "val_every": 0,
+  "val_split": "val_cache_good",
+  "val_max_samples": null,
+  "save_best_val": true,
+  "no_save_best_val": false,
+  "min_ram_gb": 0,
+  "batch_size": 2,
+  "num_workers": 8,
+  "prefetch_factor": 2,
+  "no_persistent_workers": false,
+  "lr": 0.0003,
+  "lr_scheduler": "cosine",
+  "lr_warmup_epochs": 1,
+  "lr_min": 1e-06,
+  "lr_step_size": 10,
+  "lr_gamma": 0.5,
+  "grad_accum": 1,
+  "grad_clip": 1.0,
+  "run_name": "gasa_E_box_50s",
+  "run_subdir": "rebuttal",
+  "checkpoint_dir": "/home/nvidia/mv_sam3/checkpoints",
+  "config": null,
+  "resume": null,
+  "load_weights": null,
+  "focal_weight": 2.0,
+  "focal_alpha": 0.25,
+  "all_query_supervision": false,
+  "no_object_bce_weight": 0.1,
+  "focal_gamma": 2.0,
+  "dice_weight": 0.5,
+  "boundary_weight": 0.0,
+  "lovasz_weight": 0.0,
+  "use_point_sampling": false,
+  "num_sample_points": 4096,
+  "loss_at_native_res": false,
+  "attn_map_size": 64,
+  "seed": 42,
+  "use_presence_token": true,
+  "use_box_prompts": true,
+  "box_prompt_dropout": 0.2,
+  "use_point_prompts": true,
+  "point_prompt_dropout": 0.2,
+  "num_pos_points": 10,
+  "num_neg_points": 2,
+  "samples_per_scene": 1,
+  "min_category_samples": 1,
+  "exclude_categories": null,
+  "prompt_type": "all",
+  "use_world_pe": true,
+  "use_gasa": true,
+  "gasa_beta_init": 1.0,
+  "gasa_kernel_dim": 32,
+  "gasa_fixed_kernel": false,
+  "gasa_kernel_type": "multiplicative",
+  "gasa_kernel": "multiplicative",
+  "gasa_mono_aux_weight": 0.0,
+  "gasa_kernel_lr_mult": 20.0,
+  "gasa_bidirectional": false,
+  "use_image_to_token": false,
+  "use_pos_refine": false,
+  "per_query_anchor": true,
+  "use_box_rpb": false,
+  "dim_feedforward": 2048,
+  "post_norm": true,
+  "ffn_fp32": true,
+  "use_query_pe": false,
+  "use_depth_crossattn": false,
+  "per_layer_text": true,
+  "no_initial_text": false,
+  "no_text_proj": false,
+  "clean_v": false,
+  "additive_pe": true,
+  "cross_view": false,
+  "use_da3_poses_for_gasa": false,
+  "use_gt_poses_for_gasa": false,
+  "no_gt_poses": false,
+  "sheaf_use_gt_poses": false,
+  "pe_type": "world",
+  "no_pointmap_normalize": false,
+  "concat_pointmaps_to_memory": false,
+  "concat_pointmap_dim": 64,
+  "use_iterative_pos": true,
+  "presence_weight": 2.0,
+  "presence_focal": true,
+  "presence_alpha": 0.5,
+  "presence_gamma": 0.0,
+  "use_centroid_head": false,
+  "centroid_weight": 0.5,
+  "mask_based_centroid": false,
+  "use_triangulation": false,
+  "eval_localization": false,
+  "use_iou_head": false,
+  "iou_head_weight": 1.0,
+  "pred_logits_source": "mask_mean",
+  "contrastive_weight": 0.0,
+  "contrastive_margin": 0.5,
+  "contrastive_source": "logits",
+  "no_object_weight": 1.0,
+  "align_weight": 1.0,
+  "align_alpha": 0.5,
+  "align_gamma": 2.0,
+  "align_tau": 2.0,
+  "per_layer_align": false,
+  "per_layer_align_weight": null,
+  "use_sheaf_loss": false,
+  "sheaf_weight": 0.1,
+  "sheaf_type": "constant",
+  "sheaf_d_edge": 32,
+  "sheaf_threshold": 0.1,
+  "sheaf_soft_correspondences": false,
+  "sheaf_sigma": 0.1,
+  "sheaf_detach_target": true,
+  "sheaf_max_frame_distance": 0,
+  "sheaf_symmetric_detach": true,
+  "sheaf_mutual_nn": true,
+  "output_localization": false,
+  "output_depth": false,
+  "output_pointcloud": false,
+  "semantic_union": false,
+  "mask_selection": "confidence",
+  "vis_every_epochs": 4,
+  "vis_every_batches": 0,
+  "use_lora": false,
+  "lora_rank": 8,
+  "lora_alpha": 16.0,
+  "lora_sam3": false,
+  "lora_da3": false,
+  "lora_mask_embed": false,
+  "query_proj_mlp": false,
+  "no_query_proj": false,
+  "train_mask_embed": false,
+  "use_mask_refiner": false,
+  "use_sam3_box_prompt": true,
+  "sam3_box_drop_prob": 0.3,
+  "use_sam3_decoder_only": false,
+  "sam3_decoder_from_scratch": false,
+  "use_sam3_heads": true,
+  "use_sam3_loss": true,
+  "init_scoring_from_sam3": false,
+  "init_text_crossattn_from_sam3": false,
+  "init_decoder_from_sam3": false,
+  "warmstart_gasa_from_sam3": false,
+  "torch_compile": false,
+  "compile_mode": "reduce-overhead",
+  "batch_views": false,
+  "profile": false,
+  "use_spatial_tokens": false,
+  "use_spatial_attn_bias": false,
+  "use_text_spatial_bias": false,
+  "use_spatial_points": false,
+  "use_object_aware_spatial": false,
+  "spatial_augment_prob": 0.0,
+  "spatial_multi_instance_only": true,
+  "spatial_relational_prob": 0.0,
+  "spatial_gt_aware": false,
+  "spatial_ranking_weight": 0.0,
+  "spatial_ranking_margin": 0.5,
+  "mask_smooth_kernel": 0,
+  "include_categories": null,
+  "class_balanced": true,
+  "class_balance_power": 0.7,
+  "world_size": 8,
+  "resumed_from": null
+}

checkpoints/gasa_generalist/last.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f7119d10eed584797f20d109ab5513e4e11b1646b0faae7e9f37276261ae3ffb
+size 170800385

checkpoints/gasa_generalist/summary.json

@@ -0,0 +1,562 @@
+{
+  "best_iou": 0.5617605537044844,
+  "best_val_miou": 0.0,
+  "final_loss": 0.2817773302615677,
+  "final_iou": 0.5589517329937845,
+  "final_miou": 0.496942410924111,
+  "final_mAcc": 0.8710223287508633,
+  "final_mean_class_recall": 0.7457910695787293,
+  "num_categories": 547,
+  "per_category_iou": {
+    "tv": 0.7162096889251011,
+    "cooking pot": 0.48959611857970053,
+    "spices": 0.7232083263042857,
+    "wheelbarrow": 0.6269135366828226,
+    "sponge": 0.305085791877546,
+    "paper": 0.5439921039813367,
+    "note": 0.44792840915003207,
+    "window sill": 0.3320823768432042,
+    "smoke detector": 0.524838106827763,
+    "towel": 0.6922721859097463,
+    "lemons": 0.7338203401438192,
+    "air freshener": 0.3811140363031731,
+    "wifi router": 0.1483624065503738,
+    "headphone": 0.5439145165493875,
+    "lateral pulley": 0.43692819923169873,
+    "folded blanket": 0.7537903518317012,
+    "power strip": 0.3936749116607774,
+    "counter": 0.5212496234777745,
+    "picture": 0.6991560498508637,
+    "kettlebell": 0.3742489270386266,
+    "thermostat": 0.7754084741068956,
+    "extension cord": 0.25594079317834445,
+    "plank": 0.802585987359453,
+    "fume hood": 0.5097505827607772,
+    "cleaning product": 0.20669339262817912,
+    "toothpaste": 0.6614520958083833,
+    "telephone": 0.6648286068939177,
+    "janitorial trolley": 0.3670609291420843,
+    "tiled wall": 0.5532905383640568,
+    "cutting board": 0.6236272592465238,
+    "remote control": 0.7597297297297297,
+    "books": 0.43591334523061254,
+    "cable": 0.28873382699975586,
+    "rack": 0.39461031441946354,
+    "picture frame": 0.1449980281319837,
+    "document rack": 0.5323774365559182,
+    "electric kettle": 0.6762842296814365,
+    "curtain rod": 0.6999880497131931,
+    "cup": 0.3657661061779709,
+    "flatbed trolley": 0.32081994882125553,
+    "door frame": 0.2808835086122121,
+    "knife block": 0.7305699481865285,
+    "toilet paper roll": 0.3781375111972359,
+    "carpet": 0.8062580748204761,
+    "chair": 0.5542632214491451,
+    "monitor": 0.5874821863650991,
+    "drying machine": 0.8572463960001216,
+    "conference table": 0.8058235557709719,
+    "toilet paper holder": 0.5261482254697286,
+    "pen stand": 0.24960380348652933,
+    "water filter": 0.41653946208586845,
+    "chips can": 0.23274872971643992,
+    "jar": 0.48360570417403975,
+    "paper towel roll": 0.6028711726457736,
+    "mug": 0.6397750682033294,
+    "wardrobe": 0.6794586417443002,
+    "cap": 0.46302665520206365,
+    "foosball table": 0.7591064855946186,
+    "tray": 0.7114974915813346,
+    "frame": 0.37749924296365406,
+    "pen holder": 0.5269447317017121,
+    "curtain": 0.8472191391108489,
+    "cabinet": 0.5661242820962749,
+    "clothes": 0.35883164499546416,
+    "trivets": 0.21647884677170032,
+    "towel holder": 0.5035607147515325,
+    "cover up panel": 0.5836153639538006,
+    "stool": 0.33398205908130124,
+    "oven mitts": 0.6333699319586141,
+    "plate": 0.5923023882655299,
+    "foodstuffs": 0.46041469629526793,
+    "toilet paper": 0.5414227947988534,
+    "shirt": 0.20336586055720549,
+    "storage stool": 0.9013251067660588,
+    "barstool": 0.4371326291598533,
+    "air fryer": 0.20238266111669298,
+    "pool cue": 0.07832975494188199,
+    "tool box": 0.5803090814377496,
+    "headphones": 0.5341456429482674,
+    "watering can": 0.38762867814521274,
+    "plastic container": 0.49422736007865686,
+    "whiteboard stand": 0.3493315697482167,
+    "ventilation shaft": 0.3476631209246758,
+    "fruit bowl": 0.17156225630361321,
+    "container": 0.4042319318494092,
+    "blind": 0.6689193006068972,
+    "bed": 0.30538658234135657,
+    "lab bench": 0.3289299067736411,
+    "freezer display counter": 0.6848618204779622,
+    "electrical panel": 0.5817125455783948,
+    "thermos": 0.46635451129927663,
+    "beverage crate": 0.4708456177968373,
+    "soap": 0.5220119438136093,
+    "tv stand": 0.6334277923327808,
+    "ladder": 0.684676286288482,
+    "cloth hanger": 0.23238048299655004,
+    "pillow": 0.471842366345849,
+    "storage bin": 0.3173530817009078,
+    "shower tap": 0.4105371782376885,
+    "foam tile": 0.3325689176836602,
+    "shelf rail": 0.20011008501009112,
+    "shower loofah": 0.6924952115619014,
+    "water pitcher": 0.41469435167707275,
+    "blind rail": 0.18803244053704082,
+    "suspended ceiling": 0.3173028970551464,
+    "bookshelf": 0.44557241539011827,
+    "whiteboard": 0.830254520626795,
+    "bulletin board": 0.765917761763459,
+    "egg carton": 0.3956984897518878,
+    "tablet": 0.44842598794373745,
+    "paper towel": 0.7927211430769405,
+    "coffee table": 0.8783221567153662,
+    "partition": 0.29073063987581926,
+    "box": 0.37156933536963316,
+    "bar cart": 0.3581084933225011,
+    "disinfectant dispenser": 0.6442709322194797,
+    "file cabinet": 0.3687808772892859,
+    "plant": 0.34030840584735755,
+    "refrigerator": 0.8162023442781071,
+    "keyboard": 0.5626560966214789,
+    "toilet brush": 0.4040777393585311,
+    "bouquet": 0.8123168412311483,
+    "umbrella": 0.26132896115134124,
+    "frying pan": 0.3304091601685041,
+    "wire junction": 0.1316393153343921,
+    "wall clock": 0.8654603334171065,
+    "square ducts": 0.5570904359925556,
+    "bathroom sink": 0.8464201248042673,
+    "jacket": 0.6436189406254766,
+    "cooking utensils": 0.2636283726404553,
+    "study table": 0.6304349069718217,
+    "bag": 0.33341509237031625,
+    "shelf": 0.4924830489305241,
+    "cardboard": 0.37171611831088613,
+    "standing mirror": 0.13506686384148162,
+    "guitar case": 0.04284591194968553,
+    "fan": 0.3592086004138697,
+    "sofa chair": 0.5765993874942729,
+    "tooth cleaning product": 0.30505175726771216,
+    "rice cooker": 0.7989286503203005,
+    "folder": 0.3517131155728698,
+    "food container": 0.5750047257703006,
+    "television stand": 0.16137339055793992,
+    "iron": 0.3498735777496839,
+    "broom": 0.40386298084713357,
+    "pipe": 0.3621756324275842,
+    "air vent": 0.613794877658358,
+    "exit sign": 0.7706048615036744,
+    "kettle": 0.7185433040230332,
+    "range": 0.736975396964854,
+    "ping pong paddle": 0.3445767195767196,
+    "structural column": 0.8671644991538586,
+    "water jug": 0.40443397675490245,
+    "tissue box": 0.3142925471037393,
+    "slipper": 0.4465560344339996,
+    "ironing board": 0.49232394704403926,
+    "bean bag": 0.9521502176694143,
+    "radio": 0.41724380478303214,
+    "hanger": 0.4387676660162025,
+    "pool cue holder": 0.1447601502161128,
+    "toilet flush button": 0.6825232874375764,
+    "soap dispenser": 0.6492639369318639,
+    "jerry can": 0.9030405385172808,
+    "vent": 0.061415683553088136,
+    "suitcase": 0.7663092510903323,
+    "dishcloth": 0.5528745644599303,
+    "shower wall": 0.5064927960892339,
+    "ping pong table": 0.7394266760021376,
+    "tennis ball": 0.5258064516129032,
+    "pot": 0.5891522280813583,
+    "shampoo bottle": 0.4736463133640553,
+    "notebook": 0.4065262721474996,
+    "step ladder": 0.18112879164594728,
+    "shower curtain rod": 0.7160943318163012,
+    "spray bottle": 0.3425522148267156,
+    "coffee pot": 0.6593722670294432,
+    "dish drying rack": 0.20316622691292877,
+    "wall lamp": 0.5872218798757064,
+    "bucket": 0.5327467996356139,
+    "alarm": 0.7994173949230129,
+    "shoe": 0.2601469215271977,
+    "handle": 0.17803030303030304,
+    "blackboard": 0.8706925076813992,
+    "wine bottles": 0.2970996831586644,
+    "conduit": 0.5087584431093033,
+    "wooden crate": 0.40098198229963683,
+    "bottle": 0.4479765030021054,
+    "binders": 0.6012959261046391,
+    "cleaning trolley": 0.6084065044242524,
+    "lab appliance": 0.3164478393980473,
+    "dishwasher": 0.8245447236437423,
+    "coat hanger": 0.12439591078066914,
+    "opaque window panel": 0.4902502919963311,
+    "bread box": 0.2547842019543974,
+    "tap": 0.44390452495723837,
+    "cover": 0.3380694804205394,
+    "exhaust fan": 0.4244928010471204,
+    "magazine": 0.7277261482917697,
+    "blinds": 0.7729832864158224,
+    "stove top": 0.584725788862701,
+    "hole puncher": 0.48184176394293127,
+    "cream bottle": 0.09957825679475164,
+    "microwave": 0.7363166600793407,
+    "shower curtain": 0.8707727038406184,
+    "teakettle": 0.24504041146216018,
+    "soap bar": 0.33676573835712614,
+    "neon light": 0.5738782162424578,
+    "safety bollard": 0.38150298746107886,
+    "fire extinguisher": 0.6448937803941643,
+    "toilet paper stack": 0.6819348116816472,
+    "whiteboard eraser": 0.5043846386452978,
+    "mattress": 0.5160868483491411,
+    "ceiling light": 0.628983353447502,
+    "fuse box": 0.5954549124805986,
+    "multifunction printer": 0.884639561771771,
+    "photo": 0.8311373245317183,
+    "weight": 0.5181770602681267,
+    "plush toy": 0.48932676518883417,
+    "bread": 0.402514030906435,
+    "messenger bag": 0.45382443450398147,
+    "toilet brush holder": 0.48120973514674303,
+    "dish soap bottle": 0.42775436484045753,
+    "coffee maker": 0.7025721989276524,
+    "hanging storage bin": 0.45418884841692353,
+    "ribbon": 0.4324225997744388,
+    "storage box": 0.5264154741426637,
+    "cereal box": 0.4724667118032047,
+    "coat rack": 0.2834790915229672,
+    "bathtub tap": 0.4715455636117052,
+    "decoration lamp": 0.30031612223393045,
+    "cardboard box": 0.42132270044432657,
+    "wall hooks": 0.6661754704148719,
+    "coffee machine": 0.8508472349927755,
+    "cart": 0.48288935059937094,
+    "switch": 0.2674731182795699,
+    "lab cabinet": 0.5348840924066051,
+    "painting": 0.5787677589968605,
+    "poster": 0.5907444668008048,
+    "decoration": 0.2925967933185544,
+    "pedestal fan": 0.6393399638336347,
+    "beverage crates": 0.5227947261541939,
+    "shower hose": 0.34220980598200784,
+    "dumbbell": 0.49521863506334285,
+    "shoe rack": 0.08053996909815402,
+    "bathroom mat": 0.7123623189462251,
+    "faucet": 0.6107615772916819,
+    "phone charger": 0.38605619146722164,
+    "tube": 0.1187597707139135,
+    "toilet cabinet": 0.2721411270839296,
+    "bench cushion": 0.20316956392685226,
+    "power outlet": 0.452893138133652,
+    "desk organizer": 0.1941387334129929,
+    "guillotine paper cutter": 0.8987796249922938,
+    "junction box": 0.4855612569759167,
+    "paper towel dispenser": 0.6790721860085056,
+    "rolling cart": 0.32343376939912455,
+    "speaker": 0.21190193325413112,
+    "bowl": 0.5064814075434534,
+    "foam tiles": 0.40285101523129635,
+    "lab stool": 0.42875736986046653,
+    "book": 0.22040879053327186,
+    "toilet": 0.840616836889133,
+    "shower": 0.4960127591706539,
+    "blanket": 0.8048050661382745,
+    "wine rack": 0.5071691081318909,
+    "backpack": 0.548619582191579,
+    "ceiling light fixture": 0.6283269553252784,
+    "stair": 0.6697574652948578,
+    "light": 0.47999033252099654,
+    "wood planks": 0.5801675953596928,
+    "boot": 0.7378183831672204,
+    "scale": 0.49648199248541375,
+    "detergent bottle": 0.3013517547226131,
+    "spice rack": 0.2935460262387509,
+    "printer": 0.7999548889973525,
+    "curtain rail": 0.37963609853892655,
+    "toy car": 0.22955326460481099,
+    "toilet flush": 0.7703075327871655,
+    "wooden basket": 0.4338122843831134,
+    "trash can": 0.6470320482490552,
+    "laptop case": 0.780271430856496,
+    "flip flops": 0.145306258322237,
+    "ceiling fan": 0.5299078323818053,
+    "rolled paper": 0.34960740889873165,
+    "glove dispenser": 0.6757021845742309,
+    "air conditioner": 0.6590224794644685,
+    "toothpaste bottle": 0.49125397546569743,
+    "office chair": 0.5657158726790172,
+    "hand soap dispenser": 0.48459494601164566,
+    "trivet": 0.5740479548660085,
+    "stove": 0.43523316062176165,
+    "marker": 0.17848970251716248,
+    "lunch bag": 0.598821638993037,
+    "banana": 0.8186315995052406,
+    "shoe box": 0.6183888654576128,
+    "cables": 0.23021418914110867,
+    "apple": 0.8524100298817722,
+    "basket": 0.5306658905922887,
+    "bathroom box": 0.6593651350403367,
+    "bathroom cabinet drawer": 0.5085893714515216,
+    "bathroom shelf": 0.4718469290878551,
+    "bedside lamp": 0.23625478927203064,
+    "bench": 0.5816189326075594,
+    "board": 0.5475272582391991,
+    "brush": 0.2073268277075668,
+    "cable container": 0.6510037148164023,
+    "cable duct": 0.3336653927659343,
+    "cable raceway": 0.02921517085993241,
+    "calendar": 0.9082167516176842,
+    "candle": 0.6154341664519454,
+    "canister": 0.1095450739603684,
+    "charger": 0.5136958017894012,
+    "clock": 0.3266194597429845,
+    "closet": 0.8745408787252379,
+    "cloth": 0.03931203931203931,
+    "computer": 0.32565500171156225,
+    "computer tower": 0.36837357757424366,
+    "crate": 0.34674723948706654,
+    "desk chair": 0.28035249349088726,
+    "desk lamp": 0.4716341092525442,
+    "dish sponge": 0.12841892824897763,
+    "dishes": 0.34888262532715925,
+    "dishrag": 0.6486034027840961,
+    "document binder": 0.2789125431530495,
+    "document folder": 0.35580488732528287,
+    "door stopper": 0.04660700969425802,
+    "door window": 0.39962311557788943,
+    "drone": 0.31533477321814257,
+    "drum": 0.9101748675623162,
+    "electric guitar": 0.7376257352823963,
+    "electric kettle base": 0.40719030871434153,
+    "electric wiring": 0.506587070249354,
+    "electrical duct": 0.3018555612862986,
+    "fabric": 0.6551027062275839,
+    "file folder": 0.49306370070778566,
+    "file organizer": 0.22576049935896605,
+    "first aid box": 0.7645207439198856,
+    "first aid chart": 0.5320991973875335,
+    "floor cleaner": 0.28211567586220115,
+    "flower pot": 0.3775751564169083,
+    "foldable hanging closet": 0.6799731803432157,
+    "folding fan": 0.19307683311893345,
+    "french window": 0.2048717375829406,
+    "gloves": 0.6481218133259339,
+    "hand towel": 0.8507672469374598,
+    "handbag": 0.4239618406285073,
+    "hanging shelf": 0.6123019233095439,
+    "hanging towel": 0.8624253615602342,
+    "hat": 0.5919538261286565,
+    "heat pipe": 0.07116166482573852,
+    "helmet": 0.8768556207350653,
+    "intercom": 0.16862105725214774,
+    "joystick": 0.34195255162997096,
+    "kitchen cabinet": 0.5641429769911017,
+    "kitchen counter": 0.5557968816546938,
+    "kitchen tiled wall": 0.7376208791363384,
+    "lab coat": 0.3150589154619671,
+    "lamp": 0.37991880074953155,
+    "laptop": 0.773888545456388,
+    "laptop charger": 0.05230732060884272,
+    "laundry basket": 0.6247932232744552,
+    "laundry drying rack": 0.2143050755573626,
+    "lid": 0.6174806499796316,
+    "light dimmer": 0.07273820258491133,
+    "lock": 0.0202122283981809,
+    "luggage": 0.7771518702118071,
+    "medical stool": 0.08383924539661833,
+    "microphone": 0.7099586966498393,
+    "mirror lamp": 0.7738846465489724,
+    "mop": 0.4448046018991965,
+    "mouse": 0.48496880317640384,
+    "napkin holder": 0.5126358053849788,
+    "office visitor chair": 0.30018457504614376,
+    "olive oil bottle": 0.4603509363927788,
+    "oven": 0.6923119351356463,
+    "package": 0.20604395604395603,
+    "paper shredder": 0.31714060541853467,
+    "paper tray": 0.27426451319255757,
+    "pen": 0.2677595628415301,
+    "phone": 0.44297520661157025,
+    "photocopy machine": 0.7134292090627313,
+    "pinboard": 0.5398062807923049,
+    "plastic basket": 0.6600411584600713,
+    "plug": 0.5546825228286261,
+    "plushie": 0.7983948022167017,
+    "pool table": 0.9416996618776474,
+    "projector": 0.6675863787671111,
+    "projector holder": 0.5902552204176335,
+    "range hood": 0.6795188411770376,
+    "refillable water bottle": 0.7915020945541592,
+    "rolled poster": 0.44413470956076245,
+    "rolling mat": 0.440646492434663,
+    "roman chair": 0.772459766043265,
+    "router": 0.382680639823497,
+    "rug": 0.7092252034971359,
+    "sack": 0.18142011834319527,
+    "scarf": 0.38581467617769616,
+    "scissor": 0.5735516761934188,
+    "scrubber": 0.21802982376863986,
+    "service trolley": 0.43512141618202227,
+    "shopping bag": 0.6570416392216901,
+    "shower curtain rail": 0.5832944897619444,
+    "shower faucet": 0.2522770273243279,
+    "shower glass": 0.4987381007338039,
+    "sink": 0.7698504714926149,
+    "sliding window": 0.09657305270011395,
+    "sofa module": 0.6624033369899376,
+    "spice bottles": 0.6457667258732979,
+    "standing lamp": 0.5037166192530054,
+    "storage cabinet": 0.7054001101144192,
+    "storage shelf": 0.3004010754954368,
+    "storage trolley": 0.4753785801076202,
+    "stuffed toy": 0.44430651706851276,
+    "table": 0.5262516869801426,
+    "table cloth": 0.2207336960663283,
+    "tennis racket": 0.3178681677864536,
+    "toaster": 0.3779277183185978,
+    "tools": 0.3146648555456873,
+    "toothbrush": 0.40432601030648224,
+    "trash bag": 0.31048853842203034,
+    "tripod": 0.1732706514439221,
+    "vacuum cleaner": 0.5999467571794616,
+    "vase": 0.33536087827312516,
+    "wall mirror": 0.176774445067128,
+    "webcam": 0.42806023557477263,
+    "window lintel": 0.5346481288662951,
+    "wine bottle": 0.2583024825783972,
+    "wire tray": 0.6347422134501517,
+    "wood branch": 0.6258707430340558,
+    "yoga mat": 0.5857578768095373,
+    "air duct": 0.6701720977500024,
+    "air purifier": 0.8697784235379586,
+    "bathroom cabinet": 0.6219232804756905,
+    "bathroom ceiling light": 0.6124556978192152,
+    "bathroom hook": 0.15800788247939807,
+    "bathtub": 0.8745182355699402,
+    "bathtub grab rail": 0.756456820016142,
+    "bed frame": 0.1687251919163426,
+    "boiler": 0.541011844427814,
+    "clothes drying rack": 0.2972904221818724,
+    "clothes iron": 0.373635027299454,
+    "cooking utensil": 0.31418156570039923,
+    "cupboard": 0.397638404157151,
+    "dish soap": 0.37470377514466796,
+    "drying rack": 0.7164167324388319,
+    "duster": 0.7930243230839835,
+    "fake ceiling": 0.6493741325381625,
+    "file holder": 0.5126658020561403,
+    "fireplace surround": 0.7378006724717648,
+    "flower": 0.8914881119595475,
+    "footstool": 0.8442841197125782,
+    "glasses case": 0.2240544431357152,
+    "grandfather clock": 0.9173007299320266,
+    "hand bag": 0.3257953525481201,
+    "hand dryer": 0.330570313524859,
+    "heater": 0.6927923134866913,
+    "kitchen shelf": 0.5037589704861045,
+    "lab incubator": 0.3328010657517747,
+    "light switch": 0.6228050800662618,
+    "mobile phone": 0.4271255060728745,
+    "palm rest": 0.45616201531242284,
+    "pan": 0.17701342281879195,
+    "paper bag": 0.45763555266038775,
+    "paper roll": 0.5220855136001509,
+    "phone tripod": 0.39923712650985377,
+    "pillar": 0.7347974664530342,
+    "plant pot": 0.45838763220345957,
+    "pulley": 0.45705112672643566,
+    "radiator": 0.5909387395328338,
+    "rags": 0.838929240327075,
+    "reusable plastic bag": 0.8552411851710482,
+    "reusable shopping bag": 0.6102352028948048,
+    "shelf contents": 0.5514968856741008,
+    "shower rail": 0.16522268262293183,
+    "side table": 0.7512896161016489,
+    "sign": 0.45411566061629377,
+    "slide bar": 0.39049847291882656,
+    "smoke alarm": 0.5915747901925292,
+    "spice tray": 0.3612872915198497,
+    "stall door": 0.6880878610436618,
+    "step stool": 0.3880900154358599,
+    "stick": 0.36407517715928933,
+    "storage rack": 0.4243322866373247,
+    "structural pillar": 0.251216583454465,
+    "table lamp": 0.6710157367668097,
+    "tempering valve": 0.5816214114444949,
+    "tiled floor": 0.9823025272861813,
+    "toilet seat": 0.8350859174810269,
+    "treadmill": 0.7663398932964918,
+    "wet wipes": 0.6369550812232686,
+    "acoustic guitar": 0.4728215923614766,
+    "alcoholic drinks": 0.3540532130076241,
+    "backsplash": 0.7505051396507917,
+    "beverage bottle": 0.4092319627618309,
+    "cable tray": 0.2186192468619247,
+    "calculator": 0.5908090724441435,
+    "coax outlet": 0.14538558786346398,
+    "coax outlets": 0.4463171036204744,
+    "desk": 0.6010762010927156,
+    "dining table": 0.6892236909465614,
+    "kitchen towel": 0.6806768819514057,
+    "liquid soap dispenser": 0.1629260182876143,
+    "mat": 0.7964163418258566,
+    "mirror shelf": 0.5493016906437047,
+    "nightstand": 0.4921956295525494,
+    "notice board": 0.7544732662784542,
+    "open storage cabinet": 0.5979912705787407,
+    "plastic bag": 0.7716479200142832,
+    "pole stand": 0.5392664732697886,
+    "power adapter": 0.2817124735729387,
+    "shoe mat": 0.24787685774946921,
+    "shower caddy": 0.40541757136048295,
+    "shower head": 0.594047619047619,
+    "stair railing": 0.4241948902832184,
+    "stationery": 0.2826825127334465,
+    "teapot": 0.6216030271757826,
+    "tote bag": 0.5543833331688494,
+    "towel rack": 0.3444759938048529,
+    "towel radiator": 0.5346141782325765,
+    "washing machine": 0.6967206423726215,
+    "water kettle": 0.572647051594032,
+    "binding machine": 0.2451122181155927,
+    "carrots": 0.15813953488372093,
+    "dish towel": 0.3189039465250435,
+    "gas cylinder": 0.7859057584859348,
+    "hood": 0.4533640401381806,
+    "ice cube tray": 0.30043859649122806,
+    "jug": 0.6591577563237482,
+    "kitchen drawer": 0.3896988330192734,
+    "knife": 0.5342465753424658,
+    "satchel bag": 0.621384750219106,
+    "shower cap": 0.7289719626168224,
+    "soap dish": 0.7772640404414031,
+    "sofa": 0.7557305856433545,
+    "stationery container": 0.730968745947348,
+    "table mirror": 0.10871763754045308,
+    "waffle maker": 0.6438751000800641,
+    "sleeping bag": 0.8246566396604151,
+    "stand": 0.3631660631143301,
+    "water bottle": 0.19255050505050506,
+    "clothes basket": 0.37526877930135005,
+    "floor lamp": 0.7488509150382089,
+    "network outlet": 0.17189471786551289,
+    "toilet paper dispenser": 0.45606192760575665,
+    "camera": 0.6328467153284671,
+    "clothes bin": 0.9160479838217879,
+    "mini shelf": 0.25245224574083636
+  },
+  "epochs": 20,
+  "world_size": 8,
+  "status": "completed"
+}