Assistance Requested: Issues Encountered with train.py Script in CREStereo Repository

[T-matsubara]

Hello,

I hope this message finds you well. I am currently working on a project that involves the train.py script from the CREStereo repository. However, I have encountered some issues while running the script and would like to seek assistance from the community.

The challenges I am facing are as follows:

Issue with Batch Size: The train.py script only seems to work with a batch size of 1. When attempting to use a batch size other than 1, the script fails to execute properly. I would appreciate guidance on how to make the script compatible with different batch sizes.

2RuntimeError: cuda error 700: an illegal memory access was encountered (cudaMemcpyAsync( device_ptr, host_ptr, size, cudaMemcpyHostToDevice, m_env.cuda_env().stream) at ../../../../../../src/core/impl/comp_node/cuda/comp_node.cpp:copy_to_device:230)

If anyone in the community has experience working with the train.py script and has successfully addressed these issues, I kindly request your guidance and assistance. Any insights regarding the correct configurations, dependencies, or steps needed to overcome these challenges would be greatly appreciated.

Thank you for your attention and support. I am eagerly looking forward to hearing from you or anyone who can provide valuable assistance in resolving these issues.