MCPcopy Create free account
hub / github.com/AutodeskAILab/BRepNet / brepnet_step

Method brepnet_step

models/brepnet.py:608–675  ·  view source on GitHub ↗

A train or validation step for the BRepNet network on one batch

(self, batch, batch_idx, save_segmentation_output)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 3

training_stepMethod · 0.95
validation_stepMethod · 0.95
test_stepMethod · 0.95

Calls 5

create_face_embeddingsMethod · 0.95
save_logitsMethod · 0.95
save_embeddingsMethod · 0.95
find_lossMethod · 0.95
find_predicted_classesMethod · 0.95

Tested by 1

test_stepMethod · 0.76