Comments (5)
In addition, I train the model myself and get the weight model Pt, then use the following command:
python test. py --infer_ overlap=0.5 --data_ dir=/data1/dataset/wdDATASET/BTCV/abdomen/ --pretrained_ dir='./ pretrained_ models/' --saved_ checkpoint=ckpt --pretrained_ model_ name=model. pt
The result produced an error:
Unexpected key(s) in state_ dict: "epoch", "best_acc", "state_dict".
I will --saved_ checkpoint=torchscript
The result produced an error:
PytorchStreamReader failed locating file constants. pkl: file not found
Excuse me, how to solve this? Thank you very much!!
from research-contributions.
Hello,
I faced the same issue, and I simply think they did not provide either the version with the k-fold nor the code for ensemble evaluation. In addition, I believe ensemble evaluation is fundamental to achieve high performance.
from research-contributions.
Hi @BelieferQAQ , which repository did you try ?
from research-contributions.
Hi @BelieferQAQ , which repository did you try ?
UNETR,
Using the default values for hyper-parameters, the following command can be used to initiate training using PyTorch native AMP package:
python main.py
--feature_size=32
--batch_size=1
--logdir=unetr_test
--fold=0
--optim_lr=1e-4
--lrschedule=warmup_cosine
--infer_overlap=0.5
--save_checkpoint
--data_dir=/dataset/dataset0/
from research-contributions.
@ahatamiz Hi,
Thank you for your great work. Could you please help how I can do k-fold validation using the commands? I am facing an error saying --fold in not recognized. I am using a private dataset and not a pre-trained one.
Thanks.
from research-contributions.
Related Issues (20)
- Swin-UNETR pretraining model weights do not load into the model
- Validation
- One problem came when I was using SwinUNETR to train my own data HOT 5
- Invalid organ segmentation nii file produced from prostate_mri_lesion_seg_app
- SwinUNETR-V2
- About Test HOT 4
- about evaluation HOT 1
- About SwimUNETR Model Structure
- Unnecessary conv3 Layer Weights in DAE with MONAI v0.8.0 Leading to Compatibility Issues with v0.9.0+
- FileNotFoundError in swinunetr
- Swin-UNETR pretraining validation criterion
- AttributeError: 'NoneType' object has no attribute 'items' in Auto3dSeg
- Not reproducible results
- AttributeError: 'NoneType' object has no attribute 'items' in swinunetr algo
- Brats2020 HOT 3
- Evaluation
- difference brats2021 train test
- Unexpected key(s) in state_ dict: "epoch", "best_acc", "state_dict". HOT 3
- N foreground 0, N background 52003401,unable to generate class balanced samples. NaN or Inf found in input tensor.
- Understanding the contrastive loss implementation HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from research-contributions.