mikgroup / ufloss Goto Github PK

when running the scripts "launch_training_MoDL_traditional_UFLoss_256_demo.sh" , which type of the dir_data should be uesd?
it is the data had been processed (1_0.h5) or the raw data I downloaded from "fastMRI.org" (file100001.h5)?

Traceback (most recent call last):
File "../train_ufloss.py", line 803, in
main(args)
File "../train_ufloss.py", line 562, in main
model_re.load_state_dict(
File "/home/img/anaconda3/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1223, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for Model:
size mismatch for memory_bank: copying a param with shape torch.Size([256, 1]) from checkpoint, the shape in current model is torch.Size([256, 2457600]).

This is the error when running launch_training_MoDL_traditional_UFLoss_256_demo.sh. The model shape is not corresponding, so why? I can’t deal with it.
And the other problem is in the file train_ufloss.py in line 193/194.
if args.loss_normalized == False:
output = output * std + mean
target = target * std + mean
Both the std and mean are not defined. What should I do?

I got a question about the training time is too long and the GPU memory is too high. I use GTX 3090 24G, but the batch size I can set just 1, if setting 2 I'll get cuda out of memory. So I wanna ask to how I can speed up my training time. About the GPU memory, whether the input image can be blocked and then entered to increase the batch size. Can you share some details about training? Thank you so much.

when running the scripts "launch_training_MoDL_traditional_UFLoss_256_demo.sh" , which type of the dir_data should be uesd?
it is the data had been processed (1_0.h5) or the raw data I downloaded from "fastMRI.org" (file100001.h5)?

Hello, is the visualiztion script available? How do I make sure UFLoss training is successful?

I have got some difficulties in pre-calculating the sensitive map, when runnning the code Get-patch. And I want to know how you deal with the raw data(Download from fastmri.org) , using ESPIRiT and BART. It would be better if you can email me the pre-trained model or processed data. There is my email [email protected]

i'm wondering that if your UFLoss model can use in other dataset,also knee dataset. my dataset is MICCAI13_SegChallenge.

There are some problems when I run data_preprocessing.py using single-coil data. The program will report an error : (Calibration region not found!)

/home/img/anaconda3/bin/python /home/img/Desktop/lff/UFLoss-main/UFloss_training/data_preprocessing.py
fatal: Needed a single revision
cuda:0
the last dimension 372, coil dimensional: 640: 0%| | 0/199 [00:00<?, ?it/s]ERROR: Calibration region not found!
Aborted (core dumped)
the last dimension 372, coil dimensional: 640: 0%| | 0/199 [06:34<?, ?it/s]
Traceback (most recent call last):
File "/home/img/Desktop/lff/UFLoss-main/UFloss_training/data_preprocessing.py", line 82, in
sens = bart.bart(
File "/home/img/Desktop/lff/UFLoss-main/UFloss_training/bart_/bart.py", line 94, in bart
raise Exception("Command exited with an error.")
Exception: Command exited with an error.

Process finished with exit code 1

I don’t know why ERROR:Calibration region not found and how to deal with it, so, please.
And I have another request if it's possible, may I get the code that you have trained. This will be of great help to my study.
This is my email [email protected] Thanks for the help.