| 88 | } |
| 89 | |
| 90 | static float tolerance_for(const std::string & name) { |
| 91 | if (name == "ddec_rpb_mask_0") { |
| 92 | return 2e-4f; |
| 93 | } |
| 94 | if (name == "seg_mask_embed") { |
| 95 | return 2e-4f; |
| 96 | } |
| 97 | if (name == "seg_mask_logits") { |
| 98 | return 3e-3f; |
| 99 | } |
| 100 | if (name == "ddec_layer2_out" || |
| 101 | name == "ddec_layer3_out" || |
| 102 | name == "ddec_layer4_out" || |
| 103 | name == "ddec_layer5_out") { |
| 104 | return 2e-4f; |
| 105 | } |
| 106 | return 1e-4f; |
| 107 | } |
| 108 | |
| 109 | int main(int argc, char ** argv) { |
| 110 | if (argc < 4) { |