Open zx1292982431 opened 6 months ago
model.channels
and model.arch.dim_input
to suit the channels of the dataset, where model.arch.dim_input=2*len(model.channels)
python SharedTrainer.py fit --config=configs/SpatialNet.yaml --config=configs/datasets/whamr.yaml --model.channels=[0,1] --model.arch.dim_input=4 --model.arch.dim_output=4 --model.arch.num_freqs=129 --trainer.precision=bf16-mixed --model.compile=True --data.batch_size=[2,4] --trainer.devices=0, --trainer.max_epochs=100
It worked, Thank you so much for your kindness and support. I truly appreciate it!🌹
Happy to help
I tried to train SpatialNet on WHAMR! dataset by the script
python SharedTrainer.py fit --config=configs/SpatialNet.yaml --config=configs/datasets/whamr.yaml --model.arch.dim_input=12 --model.arch.dim_output=4 --model.arch.num_freqs=129 --trainer.precision=bf16-mixed --model.compile=True --data.batch_size=[2,4] --trainer.devices=0,1,2,3, --trainer.max_epochs=100
, but I got an error:May I ask if you have encountered the similar problem and how to fix it?