Skip to content

Commit 3b5442e

Browse files
authored
config(default): default supervise flag to true.
Update pl_model.py
1 parent 188cc9e commit 3b5442e

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

scripts/pl_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def on_validation_epoch_end(self):
150150
if self.av2_mode == 'test':
151151
print(f"\nModel: {self.model.__class__.__name__}, Checkpoint from: {self.load_checkpoint_path}")
152152
print(f"Test results saved in: {self.save_res_path}, Please run submit to zip the results and upload to online leaderboard. You processed to {self.leaderboard_version} version.")
153-
output_file = zip_res(self.save_res_path, leaderboard_version=self.leaderboard_version)
153+
output_file = zip_res(self.save_res_path, leaderboard_version=self.leaderboard_version, is_supervised = self.supervised_flag)
154154
# wandb.log_artifact(output_file)
155155
return
156156

@@ -270,4 +270,4 @@ def on_test_epoch_end(self):
270270
print(f"\n\nModel: {self.model.__class__.__name__}, Checkpoint from: {self.load_checkpoint_path}")
271271
print(f"We already write the flow_est into the dataset, please run following commend to visualize the flow. Copy and paste it to your terminal:")
272272
print(f"python tests/scene_flow.py --flow_mode '{self.vis_name}' --data_dir {self.dataset_path}")
273-
print(f"Enjoy! ^v^ ------ \n")
273+
print(f"Enjoy! ^v^ ------ \n")

0 commit comments

Comments
 (0)