Facing a problem while training

[Preyansh1112]

Dear Developers
@syaym
I am trying to train using mgmno_2000,pickle. I am running it on Google Colab, and in the last line of the train.py, I am getting a Runtime Error: Mismatch in shape: grad_output[0] has a shape of torch.Size([1]) and output[0] has a shape of torch.Size([]).
I am attaching a screenshot of the same.

The line where I am facing the error:
if name == 'main':
print("not import")
main()
else:
print("import")
pass

[MLDL-YY]

Hello! Have you solved this problem so far?

[loldeng]

Hello, are you still researching this area?

[Preyansh1112]

Hi I'm really sorry for the late response. This was my old university mail, which I don't use anymore. I'm not working in this area. But thank you for asking. Thanks and Regards Preyansh Srivastava

…

On Tue, 24 Sept, 2024, 11:27 loldeng, ***@***.***> wrote: Hello, are you still researching this area? — Reply to this email directly, view it on GitHub <#3 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AUB5GFW6NFYJH4NFQ2OSVJ3ZYEWAHAVCNFSM6AAAAABMF5EKZWVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDGNZQG42TENZWGQ> . You are receiving this because you authored the thread.Message ID: ***@***.*** .com>

-- The information contained in this electronic communication is intended solely for the individual(s) or entity to which it is addressed. It may contain proprietary, confidential and/or legally privileged information. Any review, retransmission, dissemination, printing, copying or other use of, or taking any action in reliance on the contents of this information by person(s) or entities other than the intended recipient is strictly prohibited and may be unlawful. If you have received this communication in error, please notify us by responding to this email or telephone and immediately and permanently delete all copies of this message and any attachments from your system(s). The contents of this message do not necessarily represent the views or policies of BITS Pilani.