Invalid line in transformer.py

[terateil]

Hello, I guess I found small error in the code. In line 218,

teach_tatc/src/teach/modeling/models/ET/transformer.py

Line 218 in 5c8194a

losses["object"] = object_loss * self.args.object_loss_wt

I get error that there is no such argument object_loss_wt, while training the ET model.

In the config file

teach_tatc/src/teach/modeling/models/ET/configs.py

Lines 55 to 56 in 5c8194a

	# weight of object loss
	action_aux_loss_wt = 1.0

So I guess the object_loss_wt in line 218 should be replaced with action_aux_loss_wt.
Thanks!

[aliang8]

Yes you're right, line 218 should be losses["object"] = object_loss * self.args.action_aux_loss_wt.

[terateil]

Thanks for your reply!