MCPcopy Create free account
hub / github.com/SpatialVLA/SpatialVLA / replace_compute_loss

Function replace_compute_loss

train/monkey_patch.py:328–330  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

326 return (loss, outputs) if return_outputs else loss
327
328def replace_compute_loss():
329 transformers.Trainer.compute_loss = compute_loss
330 print("Replace compute_loss!!")
331
332class SaveProcessorCallback(TrainerCallback):
333 def __init__(self, processor):

Callers 2

Calls

no outgoing calls

Tested by

no test coverage detected