image_data):"""Load image data from bytes or NumPy arrays and process it."""ifisinstance(image_data,bytes):# Load image data from bytes using rasteriowithMemoryFile(image_data)asmemfile:withmemfile.open()asdataset:ifself.band_indicesisNone:image=dataset.read()else...
chunked_param_name_to_layer_type: continue bsym = prologue_producers[pro_out_p]2 changes: 1 addition & 1 deletion 2 thunder/distributed/transforms/fsdp_v2.py Original file line numberDiff line numberDiff line change @@ -52,7 +52,7 @@ def transform_traces(self, prologue_trace, ...
( 569 self.state.fn, 570 ckpt_path, 571 model_provided=True, 572 model_connected=self.lightning_module is not None, 573 ) --> 574 self._run(model, ckpt_path=ckpt_path) 576 assert self.state.stopped 577 self.training = False File site-packages/lightning/pytorch/trainer/trainer.py:981,...