Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

could not find trainer class in nnformer.training.network_training #127

Open
apuomline opened this issue Dec 3, 2023 · 0 comments
Open

Comments

@apuomline
Copy link

Hello, author, I made the following mistakes during training. May I ask why the following errors occur?
I run the following instructions: bash train_inference.sh -c 0 -n nnformer_tumor -t 3
errors:

I am running the following nnFormer: 3d_fullres
My trainer class is: None
For that I will be using the following configuration:
num_classes: 3
modalities: {0: 'FLAIR', 1: 'T1w', 2: 't1gd', 3: 'T2w'}
use_mask_for_norm OrderedDict([(0, True), (1, True), (2, True), (3, True)])
keep_only_largest_region None
min_region_size_per_class None
min_size_per_class None
normalization_schemes OrderedDict([(0, 'nonCT'), (1, 'nonCT'), (2, 'nonCT'), (3, 'nonCT')])
stages...

stage: 0
{'batch_size': 2, 'num_pool_per_axis': [5, 5, 5], 'patch_size': array([128, 128, 128]), 'median_patient_size_in_voxels': array([138, 169, 138]), 'current_spacing': array([1., 1., 1.]), 'original_spacing': array([1., 1., 1.]), 'do_dummy_2D_data_aug': False, 'pool_op_kernel_sizes': [[2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2]], 'conv_kernel_sizes': [[3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3]]}

I am using stage 0 from these plans
I am using sample dice + CE loss

I am using data from this folder: /home/lx/project/nnFormer/dataset/nnformer_preprocessed/Task003_tumor/nnFormerData_plans_v2.1
###############################################
Traceback (most recent call last):
File "/home/lx/.conda/envs/nnFormer/bin/nnFormer_train", line 33, in
sys.exit(load_entry_point('nnformer', 'console_scripts', 'nnFormer_train')())
File "/home/lx/project/nnFormer/nnformer/run/run_training.py", line 140, in main
raise RuntimeError("Could not find trainer class in nnformer.training.network_training")
RuntimeError: Could not find trainer class in nnformer.training.network_training

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant