#test the set_train function
import mindspore.nn as nn

class Backbone(nn.Cell):
    def __init__(self):
        super().__init__()
        self.fc = nn.Dense(100, 100)
    def construct(self, x):
        x = self.fc(x)
        return x

class Head(nn.Cell):
    def __init__(self):
        super().__init__()
        self.fc = nn.Dense(100, 100)
    def construct(self, x):
        x = self.fc(x)
        return x

class Network(nn.Cell):
    def __init__(self):
        super().__init__()
        self.backbone = Backbone()
        self.head = Head()
    def construct(self, x):
        x = self.backbone(x)
        x = self.head(x)
        return x

network = Network()
network.set_train(False)
print(f"network.phase: {network.phase}\tnetwork.backbone.phase: {network.backbone.phase}\t network.head.phase: {network.head.phase}")
print(f"network.training: {network.training}\tnetwork.backbone.training: {network.backbone.training}\tnetwork.head.training: {network.head.training}")

network.phase: predict  network.backbone.phase: train    network.head.phase: train
network.training: False network.backbone.training: False        network.head.training: False